Correcting class name