update wait method of E2E scenario