Fix to polarization correction