diff --git a/proteus/__init__.py b/proteus/__init__.py
index d38fc1f4cc..08802c8d05 100644
--- a/proteus/__init__.py
+++ b/proteus/__init__.py
@@ -840,7 +840,7 @@ class Wizard(object):
         assert state in self.states
 
         self.state = state
-        while self.state != 'end':
+        while self.state != self.end_state:
             ctx = self._context.copy()
             ctx.update(self._config.context)
             if self.models: