probability bug fix