fix bug in ProbabilityPippeline