fix name of class