diff --git a/tsl/profiler/protobuf/BUILD b/tsl/profiler/protobuf/BUILD
index f127cb1f3..47698cab9 100644
--- a/tsl/profiler/protobuf/BUILD
+++ b/tsl/profiler/protobuf/BUILD
@@ -11,6 +11,9 @@ package_group(
     includes = [
         "@xla//xla/tsl/profiler:friends",
     ],
+    packages = [
+        "//third_party/py/torch_tpu/...",
+    ],
 )
 
 tf_proto_library(