Add missing state transition