Fix e2e tests