fix the class name