Fix prob calculate bug