fixing e2e tests