Fix name of extended class