correcting class names