e2e tests fix