fix small bugs
bayes fix