Fix incorrect extending class