Fix name of class