diff --git a/grain/proto/BUILD b/grain/proto/BUILD
index 6144d9441..23b4124b5 100644
--- a/grain/proto/BUILD
+++ b/grain/proto/BUILD
@@ -7,5 +7,8 @@ package(default_visibility = default_visibility)
 py_proto_library(
     name = "execution_summary_py_pb2",
     # For profiling tooling.
+    visibility = default_visibility + [
+        "//learning/gemini/cms/core/data/iterators:__subpackages__",
+        "//learning/gemini/gemax/core/data/iterators:__subpackages__",
     srcs = ["execution_summary.proto"],
 )