updated tests for distortion correction