diff --git a/packages/playwright-core/src/protocol/channels.ts b/packages/playwright-core/src/protocol/channels.ts
index 8eef17b58e..09def4da7e 100644
--- a/packages/playwright-core/src/protocol/channels.ts
+++ b/packages/playwright-core/src/protocol/channels.ts
@@ -4089,6 +4089,7 @@ export const commandsWithTracingSnapshots = new Set([
   'Page.goBack',
   'Page.goForward',
   'Page.reload',
+  'Page.expectScreenshot',
   'Page.screenshot',
   'Page.setViewportSize',
   'Page.keyboardDown',
diff --git a/packages/playwright-core/src/protocol/protocol.yml b/packages/playwright-core/src/protocol/protocol.yml
index 8e350bc7a5..3ccf0c99f6 100644
--- a/packages/playwright-core/src/protocol/protocol.yml
+++ b/packages/playwright-core/src/protocol/protocol.yml
@@ -1028,6 +1028,8 @@ Page:
         log:
           type: array?
           items: string
+      tracing:
+        snapshot: true
 
     screenshot:
       parameters: