diff --git a/src/main.py b/src/main.py
index 5d3c3ea5f616ac166c88cf6c0534ec3614a14392..dbfac3e008d04cc72f438179fdce265aa1f079ad 100644
--- a/src/main.py
+++ b/src/main.py
@@ -20,9 +20,9 @@ def main():
     else:
         if checkpoint.ok:
             loader = data.Data(args)
-            model = model.Model(args, checkpoint)
-            loss = loss.Loss(args, checkpoint) if not args.test_only else None
-            t = Trainer(args, loader, model, loss, checkpoint)
+            _model = model.Model(args, checkpoint)
+            _loss = loss.Loss(args, checkpoint) if not args.test_only else None
+            t = Trainer(args, loader, _model, _loss, checkpoint)
             while not t.terminate():
                 t.train()
                 t.test()