diff --git a/tensorflow_model_optimization/python/core/quantization/keras/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/BUILD
index 7fb0dbd9..b42fa51d 100644
--- a/tensorflow_model_optimization/python/core/quantization/keras/BUILD
+++ b/tensorflow_model_optimization/python/core/quantization/keras/BUILD
@@ -1,9 +1,6 @@
 load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
 
-package(default_visibility = [
-    "//tensorflow_model_optimization:__subpackages__",
-    "//third_party/tensorflow:__subpackages__",
-])
+package(default_visibility = ["//tensorflow_model_optimization:__subpackages__"])
 
 licenses(["notice"])