diff --git a/tools/migration/BUILD b/tools/migration/BUILD
index 1a3d1f3..c1b038e 100644
--- a/tools/migration/BUILD
+++ b/tools/migration/BUILD
@@ -27,9 +27,9 @@
     deps = [
         ":legacy_fields_migration_lib",
         "//third_party/com/github/bazelbuild/bazel/src/main/protobuf:crosstool_config_py_pb2",
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
         "@io_abseil_py//absl:app",
         "@io_abseil_py//absl/flags",
+        #internal proto upb dep,
     ],
 )
 
@@ -57,9 +57,9 @@
     python_version = "PY3",
     deps = [
         "//third_party/com/github/bazelbuild/bazel/src/main/protobuf:crosstool_config_py_pb2",
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
         "@io_abseil_py//absl:app",
         "@io_abseil_py//absl/flags",
+        #internal proto upb dep,
     ],
 )
 
@@ -70,9 +70,9 @@
     deps = [
         ":ctoolchain_comparator_lib",
         "//third_party/com/github/bazelbuild/bazel/src/main/protobuf:crosstool_config_py_pb2",
-        "//third_party/py/google/protobuf:use_fast_cpp_protos",
         "@io_abseil_py//absl:app",
         "@io_abseil_py//absl/flags",
+        #internal proto upb dep,
     ],
 )
 
diff --git a/tools/migration/ctoolchain_comparator_lib_test.py b/tools/migration/ctoolchain_comparator_lib_test.py
index c81ff47..a58d3c3 100644
--- a/tools/migration/ctoolchain_comparator_lib_test.py
+++ b/tools/migration/ctoolchain_comparator_lib_test.py
@@ -17,7 +17,7 @@
 from third_party.com.github.bazelbuild.bazel.src.main.protobuf import crosstool_config_pb2
 from tools.migration.ctoolchain_comparator_lib import compare_ctoolchains
 
-from py import mock
+from py from py import mock
 try:
   # Python 2
   from cStringIO import StringIO