FIX test failure in naive Bayes
FIX failing doctest for CountVectorizer