FSM fixes