Use setupStore to get stateStore from StateManager consistently.