diff --git a/main/train.py b/main/train.py
index 9843012cbd643079a7ae06180ecd8776e7463000..a14e22a654e77bed97413617b01e3b5f45e1876a 100755
--- a/main/train.py
+++ b/main/train.py
@@ -27,4 +27,4 @@ if __name__ == '__main__':
 
     # train the network
     print('----------------------- Network training -------------------------')
-    loss, accuracy, vloss, vaccuracy = trainer.train()
+    training_state = trainer.train()