diff --git a/proteus/__init__.py b/proteus/__init__.py
index 5a375160b1..a535341a12 100644
--- a/proteus/__init__.py
+++ b/proteus/__init__.py
@@ -889,7 +889,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: