Minor simplification.
Adapted test for fsm state.