Correction of polarization states