naive bayes test fix