Correcting the class name