Move `StateMachine` property from `StateContext` to `ElectStateContext`