diff --git a/main.py b/main.py
index 06be979f5043e6f67c09c351f15d1df539b4c89e..cd8556e94654b87dc3fbba68d36479e7e4ff12a8 100644
--- a/main.py
+++ b/main.py
@@ -34,7 +34,7 @@ def train(model, data_train, optimizer, loss_function, epoch):
         optimizer.step()
     losses = losses/len(data_train.dataset)
     acc = acc/len(data_train.dataset)
-    print('Test epoch {}, loss : {:.3f} acc : {:.3f}'.format(epoch,losses,acc))
+    print('Train epoch {}, loss : {:.3f} acc : {:.3f}'.format(epoch,losses,acc))
     return losses, acc
 
 def test(model, data_test, loss_function, epoch):