diff --git a/tsl/profiler/lib/BUILD b/tsl/profiler/lib/BUILD
index cad3c9faf..d75e7fef3 100644
--- a/tsl/profiler/lib/BUILD
+++ b/tsl/profiler/lib/BUILD
@@ -178,6 +178,7 @@ cc_library(
     visibility = internal_visibility([
         "@xla//xla/tsl:internal",
         "//third_party/py/jax:__subpackages__",
+        "@xla//xla/tsl/profiler:friends",
     ]),
     deps = [
         "//tsl/platform",