You can remove the sort that happens outside of `_count_vocab` now.