Fixing e2e test