E2E test minor refactoring