diff --git a/stateful/base.py b/stateful/base.py index 74272f4..49e8429 100644 --- a/stateful/base.py +++ b/stateful/base.py @@ -52,7 +52,7 @@ class Stateful(object): def getStateObject(self): states = self.getStatesDefinition().states - if self.state not in states: + if self.getState() not in states: self.state = self.getStatesDefinition().initialState return states[self.state]