Update controller spec for state machine