FSM for unit prep