fix test prob

fix prob