Fixes the custom e2e tests