Correct the uniform distribution