fix naive bayes