add e2e tests back