diff --git a/main/eval.py b/main/eval.py
index 3d6d521598ba9b040c89ae523a7a6f67d4692203..da3fa14c71b0175665ad76bc402c704d57db6489 100755
--- a/main/eval.py
+++ b/main/eval.py
@@ -43,7 +43,8 @@ if __name__ == '__main__':
     if trainer.plot_samples:
 
         # load pretrained model
-        state = trainer.model.load(trainer.optimizer, trainer.state_file,
+        state = trainer.model.load(trainer.state_file,
+                                   trainer.optimizer,
                                    trainer.state_path)
         trainer.model.eval()
 
@@ -79,7 +80,7 @@ if __name__ == '__main__':
                                   labels,
                                   trainer.dataset.use_bands,
                                   trainer.dataset.labels,
-                                  ypred=y_pred,
+                                  y_pred=y_pred,
                                   bands=trainer.plot_bands,
                                   state=sname,
                                   stretch=True,