Correction to class