diff --git a/train.py b/train.py
index 26f0c280..92d217f7 100644
--- a/train.py
+++ b/train.py
@@ -283,7 +283,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
             #                                    ('linear_loss', linear_loss.item()),
             #                                    ('mel_loss', mel_loss.item()),
             #                                    ('stop_loss', stop_loss.item())])
-            if current_step % c.print_step == 0:
+            if num_iter % c.print_step == 0:
                 print(" | | > TotalLoss: {:.5f}   LinearLoss: {:.5f}   MelLoss:{:.5f}  "\
                       "StopLoss: {:.5f}  ".format(loss.item(),
                                                           linear_loss.item(),