Did you forget to remove the `upgrade` state from the state machine?