fix unigram bug