flaky e2e tests fix