Should this check also be in `_get_leader_for_partition()`?