diff --git a/invokeai/app/services/graph.py b/invokeai/app/services/graph.py
index 711101ad97..3e44415d3d 100644
--- a/invokeai/app/services/graph.py
+++ b/invokeai/app/services/graph.py
@@ -882,7 +882,7 @@ class GraphExecutionState(BaseModel):
                 if self.batch_indices[index] > 0:
                     self.executed.clear()
                     self.batch_indices[index] -= 1
-                    return next(self)
+                    return self.next(self)
 
         # If next is still none, there's no next node, return None
         return next_node