diff --git a/docs/examples/multitask.ipynb b/docs/examples/multitask.ipynb
index 179a2355..a31535a1 100644
--- a/docs/examples/multitask.ipynb
+++ b/docs/examples/multitask.ipynb
@@ -213,6 +213,7 @@
         "The user and movie models are as before:\n",
         "\n",
         "```python\n",
+        "embedding_dimension=32\n",
         "user_model = tf.keras.Sequential([\n",
         "  tf.keras.layers.StringLookup(\n",
         "      vocabulary=unique_user_ids, mask_token=None),\n",