diff --git a/tensorflow_probability/substrates/BUILD b/tensorflow_probability/substrates/BUILD
index 5a186f4cfa..5e45d3fcfb 100644
--- a/tensorflow_probability/substrates/BUILD
+++ b/tensorflow_probability/substrates/BUILD
@@ -48,6 +48,7 @@ py_library(
     srcs = ["jax/__init__.py"],
     tags = ["alt_dep=//tensorflow_probability:jax"],
     deps = [
+        # jax dep,
 #         "//tensorflow_probability/google:google.jax",  # DisableOnExport
 #         "//tensorflow_probability/google/autosts:autosts.jax",  # DisableOnExport
 #         "//tensorflow_probability/google/staging:staging.jax",  # DisableOnExport