Looks like this logic is duplicated in `#update_counters` as well (existing code, not in your change) that would have to be updated also.