diff --git a/tensorflow_probability/python/__init__.py b/tensorflow_probability/python/__init__.py
index d81375f6f4..75f91c6b0e 100644
--- a/tensorflow_probability/python/__init__.py
+++ b/tensorflow_probability/python/__init__.py
@@ -83,7 +83,8 @@ def _validate_tf_environment(package):
             'installed by default when you install TensorFlow Probability. '
             'This is so that JAX-only users do not have to install TensorFlow '
             'or TF-Keras. To use TensorFlow Probability with TensorFlow, '
-            'please install the tf-keras or tf-keras-nightly package.\n\n')
+            'please install the tf-keras or tf-keras-nightly package.\n'
+            'This can be be done through the tensorflow-probability[tf] extra.\n\n')
       raise