diff --git a/tensorflow_text/BUILD b/tensorflow_text/BUILD
index 236461b27..6b192e219 100644
--- a/tensorflow_text/BUILD
+++ b/tensorflow_text/BUILD
@@ -6,7 +6,10 @@ load("//tensorflow_text:tftext.bzl", "extra_py_deps", "py_tf_text_library")
 
 # Visibility rules
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//learning/gemini/prod/serving/export:__subpackages__",
+        "//visibility:public",
+    ],
 )
 
 licenses(["notice"])