Bayes classification fix