NaiveBayes+Knn bug fix