fixed regression in getProbability