Correction in code