diff --git a/docs/tutorials/ranking_dnn_distributed.ipynb b/docs/tutorials/ranking_dnn_distributed.ipynb
index 3eeb8b1..3fb6ad4 100644
--- a/docs/tutorials/ranking_dnn_distributed.ipynb
+++ b/docs/tutorials/ranking_dnn_distributed.ipynb
@@ -104,7 +104,8 @@
       "outputs": [],
       "source": [
         "!pip install -q tensorflow-ranking tensorflow-serving-api\n",
-        "!pip install -U \"tensorflow-text==2.11.*\""
+        "!pip install -U \"tensorflow-text==2.11.*\"\n",
+        "!pip install -q protobuf==3.20.3"
       ]
     },
     {