Get the probabilities right.