diff --git a/docs/tutorials/quickstart.ipynb b/docs/tutorials/quickstart.ipynb
index 84d9e03..bc00032 100644
--- a/docs/tutorials/quickstart.ipynb
+++ b/docs/tutorials/quickstart.ipynb
@@ -82,8 +82,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -q tensorflow-ranking\n",
-        "!pip install -q --upgrade tensorflow-datasets"
+        "!pip install -q tensorflow-ranking tensorflow-datasets\n",
+       
       ]
     },
     {
@@ -162,11 +162,11 @@
         "movies = movies.map(lambda x: x[\"movie_title\"])\n",
         "users = ratings.map(lambda x: x[\"user_id\"])\n",
         "\n",
-        "user_ids_vocabulary = tf.keras.layers.experimental.preprocessing.StringLookup(\n",
+        "user_ids_vocabulary = tf.keras.layers.StringLookup(\n",
         "    mask_token=None)\n",
         "user_ids_vocabulary.adapt(users.batch(1000))\n",
         "\n",
-        "movie_titles_vocabulary = tf.keras.layers.experimental.preprocessing.StringLookup(\n",
+        "movie_titles_vocabulary = tf.keras.layers.StringLookup(\n",
         "    mask_token=None)\n",
         "movie_titles_vocabulary.adapt(movies.batch(1000))"
       ]