diff --git a/dgf/src/io/BUILD b/dgf/src/io/BUILD
index bf73d9d..d2bcf8e 100644
--- a/dgf/src/io/BUILD
+++ b/dgf/src/io/BUILD
@@ -376,7 +376,7 @@ py_test(
         "//dgf/src/util:gen_test_graph",
         "//dgf/src/util:proto",
         "//dgf/src/util:test_util",
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
+        # google/protobuf:use_fast_cpp_protos dep,
         # numpy dep,
         # tensorflow dep,
         # tensorflow_gnn/proto dep,
@@ -395,7 +395,7 @@ py_test(
         "//dgf/src/data:schema",
         "//dgf/src/util:gen_test_graph",
         "//dgf/src/util:test_util",
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
+        # google/protobuf:use_fast_cpp_protos dep,
         # numpy dep,
         # tensorflow dep,
     ],
@@ -454,7 +454,7 @@ py_test(
     deps = [
         ":tfexample",
         # absl/testing:absltest dep,
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
+        # google/protobuf:use_fast_cpp_protos dep,
         # numpy dep,
         # tensorflow dep,
     ],
diff --git a/requirements.txt b/requirements.txt
index 55e4463..d62bdda 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -36,3 +36,4 @@ networkx
 pyarrow
 fastavro
 datasketch
+protobuf