Fix bug with probability being too high.