diff --git a/notebooks/examples/long_short_term_memory.ipynb b/notebooks/examples/long_short_term_memory.ipynb
index 4c9ecaa681b0be9925ac575bb626f94218896c5d..3616d25a25e859415ab488ed0743b6cf5988505e 100644
--- a/notebooks/examples/long_short_term_memory.ipynb
+++ b/notebooks/examples/long_short_term_memory.ipynb
@@ -748,11 +748,12 @@
    ],
    "source": [
     "history = None\n",
+    "epochs = 5\n",
+    "\n",
     "if LOAD_MODEL:\n",
     "  history = pd.read_csv(BASE_PATH+'lstm_training.log', sep=',', engine='python')\n",
     "else:\n",
     "  csv_logger = CSVLogger(BASE_PATH+'lstm_training.log', separator=',', append=False)\n",
-    "  epochs = 5\n",
     "  history=model.fit(x,y, epochs = epochs, verbose = 'auto', validation_data=(x_test, y_test), callbacks=[csv_logger]).history\n",
     "  model.save(BASE_PATH+'long_short_term_memory_model.h5')"
    ]