fix name of the class