fix a bit of state machine test