fixing probabilities prediction