fix up E2E tests