Correct the unit vector