fixed unigram distribution part