A state-based control system for an air handling unit (AHU) includes a finite state machine configured to transition between a high cooling load state and a low cooling load state. In the high cooling load state, the system maintains the temperature of a supply airstream provided by the AHU at a fixed setpoint and controls the temperature of a building zone by modulating the speed of a supply air fan. In the low cooling load state, the system operates the supply air fan at a fixed speed and controls the zone temperature by modulating an amount of cooling applied to the supply airstream by one or more cooling stages. A feed-forward module manages disturbances caused by adding or shedding cooling stages by applying a feed-forward gain to the supply air fan setpoint.