diff --git a/tensorflow_gnn/runner/BUILD b/tensorflow_gnn/runner/BUILD
index a82d7aa1..4fb2a3c7 100644
--- a/tensorflow_gnn/runner/BUILD
+++ b/tensorflow_gnn/runner/BUILD
@@ -28,7 +28,8 @@ pytype_strict_library(
     name = "orchestration",
     srcs = ["orchestration.py"],
     srcs_version = "PY3",
-    visibility = [":__subpackages__"],
+    # visibility = [":__subpackages__"],
+    visibility = ["//visibility:public"],
     deps = [
         "//:expect_tensorflow_installed",
         "//tensorflow_gnn",