Fixed problem with Bayes