fixing the class name