diff --git a/xla/tsl/profiler/rpc/client/BUILD b/xla/tsl/profiler/rpc/client/BUILD
index 97c55eb9ac55c..40fba67114eed 100644
--- a/xla/tsl/profiler/rpc/client/BUILD
+++ b/xla/tsl/profiler/rpc/client/BUILD
@@ -66,6 +66,7 @@ cc_library(
         "//xla/tsl/profiler/rpc:__pkg__",
         "//learning/pathways/util/platform:__subpackages__",
         "//third_party/pathways/util/platform:__subpackages__",
+        "//third_party/py/torch_tpu:__subpackages__",
     ]),
     deps = [
         "//xla/tsl/lib/io:zlib_compression_options",