diff --git a/tensorflow_model_optimization/python/core/quantization/keras/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/BUILD
index 6ad39df7..2b677e73 100644
--- a/tensorflow_model_optimization/python/core/quantization/keras/BUILD
+++ b/tensorflow_model_optimization/python/core/quantization/keras/BUILD
@@ -3,10 +3,7 @@
 
 package(
     default_applicable_licenses = ["//tensorflow_model_optimization:license"],
-    default_visibility = [
-        "//tensorflow_model_optimization:__subpackages__",
-        "//third_party/tensorflow:__subpackages__",
-    ],
+    default_visibility = ["//tensorflow_model_optimization:__subpackages__"],
 )
 
 licenses(["notice"])