diff --git a/docs/examples/basic_ranking.ipynb b/docs/examples/basic_ranking.ipynb
index f37750dd..fc53ae41 100644
--- a/docs/examples/basic_ranking.ipynb
+++ b/docs/examples/basic_ranking.ipynb
@@ -91,7 +91,8 @@
       "outputs": [],
       "source": [
         "!pip install -q tensorflow-recommenders\n",
-        "!pip install -q --upgrade tensorflow-datasets"
+        "!pip install -q --upgrade tensorflow-datasets\n",
+        "!pip intall tfds-nightly"
       ]
     },
     {