Fixed typo in the probability output part