fixing class names