Yeah, it will be good to see `SlotState` used in `persistShardState` as well.