fixed other error in prob calculation