Debugging 3 shock case: corrected the probability adjustment