Correction in class