Fix the class name