e2e test fixes