diff --git a/tf_keras/api/BUILD b/tf_keras/api/BUILD
index 903af22cd..171b9fcbf 100644
--- a/tf_keras/api/BUILD
+++ b/tf_keras/api/BUILD
@@ -24,6 +24,7 @@ py_library(
         ":tf_keras_python_api_gen_compat_v2",
     ],
     srcs_version = "PY3",
+    visibility = ["//visibility:public"],
     deps = [
         "//:expect_tensorflow_installed",
         "//tf_keras",