correct some class names