diff --git a/site/en/gemma/docs/distributed_tuning.ipynb b/site/en/gemma/docs/distributed_tuning.ipynb
index 30b8d99fc..77cb61d61 100644
--- a/site/en/gemma/docs/distributed_tuning.ipynb
+++ b/site/en/gemma/docs/distributed_tuning.ipynb
@@ -200,7 +200,7 @@
         "# Work around an import error with tensorflow-hub. The library is not used.\n",
         "!pip install -q -U tensorflow-hub\n",
         "# Install tensorflow-cpu so tensorflow does not attempt to access the TPU.\n",
-        "!pip install -q -U tensorflow-cpu\n",
+        "!pip install -q -U tensorflow-cpu tensorflow-text\n",
         "# Install keras 3 last. See https://keras.io/getting_started for details.\n",
         "!pip install -q -U keras"
       ]