Ensure state is updated when handling vote