Move `Connection` property from `StateContext` to `ElectStateContext`