diff --git a/tensorflow_model_optimization/python/core/keras/BUILD b/tensorflow_model_optimization/python/core/keras/BUILD
index 93e83b901..9954ad637 100644
--- a/tensorflow_model_optimization/python/core/keras/BUILD
+++ b/tensorflow_model_optimization/python/core/keras/BUILD
@@ -39,6 +39,7 @@ py_strict_library(
     name = "utils",
     srcs = ["utils.py"],
     srcs_version = "PY3",
+    visibility = ["//visibility:public"],
     deps = [
         # python:control_flow_ops tensorflow dep2,
         # python:smart_cond tensorflow dep2,