Fix naive bayes