diff --git a/proteus/__init__.py b/proteus/__init__.py
index 08fa4cd297..8dff002bf8 100644
--- a/proteus/__init__.py
+++ b/proteus/__init__.py
@@ -639,6 +639,8 @@ class Wizard(object):
             elif res['type'] == 'print':
                 # TODO run print
                 self.state = res['state']
+            elif res['type'] == 'state':
+                self.state = res['state']
 
         if self.state == 'end':
             self._proxy.delete(self.id, self._config.context)