diff --git a/demo.py b/demo.py
index 0bccea43a1828d755c1210e8ef4bbba168c026e4..67d826c35d78ae10ce89d3c4f81e36db3ff7624c 100644
--- a/demo.py
+++ b/demo.py
@@ -64,8 +64,7 @@ with tf.Session() as session:
 
 
     # train & validate. Allocate 20% for validation set. 100 epochs
-    history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_
-callback], verbose=1, shuffle=True, validation_split=0.2)
+    history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_callback], verbose=1, shuffle=True, validation_split=0.2)
     # plot
     plt.plot(history.history['loss'])
     plt.plot(history.history['val_loss'])