diff --git a/tensorflow/python/platform/BUILD b/tensorflow/python/platform/BUILD
index 97c88a1343c..108f7355679 100644
--- a/tensorflow/python/platform/BUILD
+++ b/tensorflow/python/platform/BUILD
@@ -186,10 +186,6 @@ py_library(
     name = "client_testlib",
     srcs = ["test.py"],
     srcs_version = "PY3",
-    visibility = visibility + [
-        # TODO(b/178183520): remove once the test no longer depend on internal files.
-        "//learning/serving/testing/prober/model_generators/tpu:__pkg__",
-    ],
     deps = [
         ":test",
         "//tensorflow/python:client",