Add wait methods in the PartnerForm class