Methods and mechanisms for correcting a wavefront error in an optical element are disclosed. A wavefront error that is downstream of an optical element in an optical path is determined. A refractive index prescription that reduces the wavefront error is determined. A beam of energy is directed at a surface of the optical element in accordance with the refractive index prescription to alter the surface to change an index of refraction at multiple locations on the surface.