diff --git a/e2e/puppeteer-custom-environment.ts b/e2e/puppeteer-custom-environment.ts
index a9f5f0a23cf..0edcc43de56 100644
--- a/e2e/puppeteer-custom-environment.ts
+++ b/e2e/puppeteer-custom-environment.ts
@@ -62,7 +62,7 @@ class PuppeteerCustomEnvironment extends PuppeteerEnvironment {
       await savePageToFile(activePage, htmlFile);
     }
     if (super.handleTestEvent) {
-      super.handleTestEvent(event, state);
+      await super.handleTestEvent(event, state);
     }
   }
 }