diff --git a/saxml/server/pax/BUILD b/saxml/server/pax/BUILD
index a399bd88..efe6a0f7 100644
--- a/saxml/server/pax/BUILD
+++ b/saxml/server/pax/BUILD
@@ -28,7 +28,7 @@ pytype_strict_library(
     name = "branch_selection",
     srcs = ["branch_selection.py"],
     srcs_version = "PY3",
-    visibility = ["//saxml:internal"],
+    visibility = ["//visibility:public"],
     deps = [
         "//third_party/py/numpy",
         "//third_party/py/tensorflow:tensorflow_no_contrib",