diff --git a/packages/rrweb/test/integration.test.ts b/packages/rrweb/test/integration.test.ts
index d2947c3cd7..32ae79d8c5 100644
--- a/packages/rrweb/test/integration.test.ts
+++ b/packages/rrweb/test/integration.test.ts
@@ -907,7 +907,7 @@ describe('record integration tests', function (this: ISuite) {
     const snapshots = (await page.evaluate(
       'window.snapshots',
     )) as eventWithTime[];
-    assertSnapshot(snapshots);
+    await assertSnapshot(snapshots);
   });
 
   it('should record images with blob url', async () => {
diff --git a/packages/rrweb/test/record.test.ts b/packages/rrweb/test/record.test.ts
index d4e0df6c79..72f945e0f7 100644
--- a/packages/rrweb/test/record.test.ts
+++ b/packages/rrweb/test/record.test.ts
@@ -534,7 +534,7 @@ describe('record', function (this: ISuite) {
       styleEl.append(document.createTextNode('span { color: orange; }'));
     });
     await waitForRAF(ctx.page);
-    assertSnapshot(ctx.events);
+    await assertSnapshot(ctx.events);
   });
 
   it('captures stylesheets with `blob:` url', async () => {
@@ -962,7 +962,7 @@ describe('record', function (this: ISuite) {
     await ctx.page.waitForResponse(corsStylesheetURL); // wait for stylesheet to be loaded
     await waitForRAF(ctx.page); // wait for rrweb to emit events
 
-    assertSnapshot(ctx.events);
+    await assertSnapshot(ctx.events);
   });
 
   it('captures adopted stylesheets in shadow doms and iframe', async () => {