diff --git a/lib/model/train_val.py b/lib/model/train_val.py
index 1b861e51..dba772be 100644
--- a/lib/model/train_val.py
+++ b/lib/model/train_val.py
@@ -203,8 +203,8 @@ def initialize(self, sess):
 
   def restore(self, sess, sfile, nfile):
     # Get the most recent snapshot and restore
-    np_paths = [sfile]
-    ss_paths = [nfile]
+    np_paths = [nfile]
+    ss_paths = [sfile]
     # Restore model from snapshots
     last_snapshot_iter = self.from_snapshot(sess, sfile, nfile)
     # Set the learning rate