diff --git a/docs/snippets/common/test-runner-image-snapshot-testing.js.mdx b/docs/snippets/common/test-runner-image-snapshot-testing.js.mdx
index df88801ddbbc..dacc97f62d5f 100644
--- a/docs/snippets/common/test-runner-image-snapshot-testing.js.mdx
+++ b/docs/snippets/common/test-runner-image-snapshot-testing.js.mdx
@@ -1,6 +1,7 @@
 ```js
 // .storybook/test-runner.js
 
+const { waitForPageReady } = require('@storybook/test-runner');
 const { toMatchImageSnapshot } = require('jest-image-snapshot');
 
 const customSnapshotsDir = `${process.cwd()}/__snapshots__`;