Add missed extends to the class