Added missing state toggle