diff --git a/16_nlp_with_rnns_and_attention.ipynb b/16_nlp_with_rnns_and_attention.ipynb
index 9cafaf9..ab41b8b 100644
--- a/16_nlp_with_rnns_and_attention.ipynb
+++ b/16_nlp_with_rnns_and_attention.ipynb
@@ -735,7 +735,8 @@
     "    return ds.map(lambda window: (window[:, :-1], window[:, 1:])).prefetch(1)\n",
     "\n",
     "stateful_train_set = to_dataset_for_stateful_rnn(encoded[:1_000_000], length)\n",
-    "stateful_valid_set = to_dataset_for_stateful_rnn(encoded[:1_000_000], length)\n",
+    "stateful_valid_set = to_dataset_for_stateful_rnn(encoded[1_000_000:1_060_000],\n",
+    "                                                 length)\n",
     "stateful_test_set = to_dataset_for_stateful_rnn(encoded[1_060_000:], length)"
    ]
   },