diff --git a/docs/_snippets/portable-stories-vitest-set-project-annotations-simple.md b/docs/_snippets/addon-vitest-set-project-annotations-simple.md
similarity index 75%
rename from docs/_snippets/portable-stories-vitest-set-project-annotations-simple.md
rename to docs/_snippets/addon-vitest-set-project-annotations-simple.md
index 296a97340c26..a38e9ae39c6d 100644
--- a/docs/_snippets/portable-stories-vitest-set-project-annotations-simple.md
+++ b/docs/_snippets/addon-vitest-set-project-annotations-simple.md
@@ -1,34 +1,22 @@
 ```tsx filename=".storybook/vitest.setup.ts" renderer="react" language="ts"
-import { beforeAll } from 'vitest';
 // Replace your-framework with the framework you are using, e.g. react-vite, nextjs, nextjs-vite, etc.
 import { setProjectAnnotations } from '@storybook/your-framework';
 import * as previewAnnotations from './preview';
 
 const annotations = setProjectAnnotations([previewAnnotations]);
-
-// Run Storybook's beforeAll hook
-beforeAll(annotations.beforeAll);
 ```
 
 ```tsx filename=".storybook/vitest.setup.ts" renderer="svelte" language="ts"
-import { beforeAll } from 'vitest';
 // Replace your-framework with the framework you are using, e.g. sveltekit or svelte-vite
 import { setProjectAnnotations } from '@storybook/your-framework';
 import * as previewAnnotations from './preview';
 
 const annotations = setProjectAnnotations([previewAnnotations]);
-
-// Run Storybook's beforeAll hook
-beforeAll(annotations.beforeAll);
 ```
 
 ```tsx filename=".storybook/vitest.setup.ts" renderer="vue" language="ts"
-import { beforeAll } from 'vitest';
 import { setProjectAnnotations } from '@storybook/vue3-vite';
 import * as previewAnnotations from './preview';
 
 const annotations = setProjectAnnotations([previewAnnotations]);
-
-// Run Storybook's beforeAll hook
-beforeAll(annotations.beforeAll);
 ```
diff --git a/docs/writing-tests/integrations/vitest-addon.mdx b/docs/writing-tests/integrations/vitest-addon.mdx
index 5d89f12cd5dc..ae7fe4241a7f 100644
--- a/docs/writing-tests/integrations/vitest-addon.mdx
+++ b/docs/writing-tests/integrations/vitest-addon.mdx
@@ -76,7 +76,7 @@ When the addon is set up automatically, it will create or adjust your Vitest con
 
   {/* prettier-ignore-start */}
 
-  <CodeSnippets path="portable-stories-vitest-set-project-annotations-simple.md" />
+  <CodeSnippets path="addon-vitest-set-project-annotations-simple.md" />
 
   {/* prettier-ignore-end */}