Fixing broken category lookup on BPE