Fixed failing e2e tests