diff --git a/main_nmt.py b/main_nmt.py
index 71d5354..24cd652 100644
--- a/main_nmt.py
+++ b/main_nmt.py
@@ -25,7 +25,7 @@
                           batch_size=len(data[1]),
                           clip= 10.0,
                           teacher_forcing_ratio=0.5,
-                          i==0)
+                          reset=i==0)
 
         pbar.set_description(model.print_loss())