Naive Bayes bug fixed