Fixed bug with probabilities in MultiClassModel