diff --git a/docs/tutorials/text_generation.ipynb b/docs/tutorials/text_generation.ipynb
index a427038df..2bc3f8e28 100644
--- a/docs/tutorials/text_generation.ipynb
+++ b/docs/tutorials/text_generation.ipynb
@@ -673,7 +673,7 @@
       "source": [
         "class MyModel(tf.keras.Model):\n",
         "  def __init__(self, vocab_size, embedding_dim, rnn_units):\n",
-        "    super().__init__(self)\n",
+        "    super().__init__()\n",
         "    self.embedding = tf.keras.layers.Embedding(vocab_size, embedding_dim)\n",
         "    self.gru = tf.keras.layers.GRU(rnn_units,\n",
         "                                   return_sequences=True,\n",