fixing class name