Make e2e tests work