diff --git a/16_nlp_with_rnns_and_attention.ipynb b/16_nlp_with_rnns_and_attention.ipynb
index 328e421..7a85926 100644
--- a/16_nlp_with_rnns_and_attention.ipynb
+++ b/16_nlp_with_rnns_and_attention.ipynb
@@ -2063,7 +2063,7 @@
     "    len(INPUT_CHARS) + 1, encoder_embedding_size)(encoder_inputs)\n",
     "\n",
     "decoder_embedding_layer = keras.layers.Embedding(\n",
-    "    len(INPUT_CHARS) + 2, decoder_embedding_size)\n",
+    "    len(OUTPUT_CHARS) + 2, decoder_embedding_size)\n",
     "decoder_embeddings = decoder_embedding_layer(decoder_inputs)\n",
     "\n",
     "encoder = keras.layers.LSTM(units, return_state=True)\n",