Fix stage checking

Becksteinlab · Sep 20, 2023 · d38deb6 · d38deb6
1 parent 3971e5d
commit d38deb6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mdpow/restart.py b/mdpow/restart.py
@@ -132,8 +132,9 @@ def completed(self, stage):
 
     def start(self, stage):
         """Record that *stage* is starting."""
-        if self.current is not None:
-            errmsg = "Cannot start stage %s because previously started stage %s " \
+        if self.current is not None and self.current != stage:
+            errmsg = (
+                "Cannot start stage %s because previously started stage %s "
                 "has not been completed." % (stage, self.current)
             logger.error(errmsg)
             raise JournalSequenceError(errmsg)