Fixed a bug on the probability distribution
Fixed a bug on the probability distribution