Fix up e2e tests.