diff --git a/RNN - Text Generator.ipynb b/RNN - Text Generator.ipynb
index 3ab8bf0..128bd13 100644
--- a/RNN - Text Generator.ipynb	
+++ b/RNN - Text Generator.ipynb	
@@ -479,14 +479,14 @@
     "batch_inputs, batch_targets = next(gen_batch(inputs, targets, 50, 64))\n",
     "\n",
     "# Make a first prediction\n",
-    "outputs = model.predict(batch_inputs)\n",
+    "outputs = model.predict(batch_inputs,batch_size=64)\n",
     "first_prediction = outputs[0][0]\n",
     "\n",
     "# Reset the states of the RNN states\n",
     "model.reset_states()\n",
     "\n",
     "# Make an other prediction to check the difference\n",
-    "outputs = model.predict(batch_inputs)\n",
+    "outputs = model.predict(batch_inputs,batch_size=64)\n",
     "second_prediction = outputs[0][0]\n",
     "\n",
     "# Check if both prediction are equal\n",