bugfix in unigram token count