Fix fsm test