diff --git a/tests/storybook/.storybook/main.ts b/tests/storybook/.storybook/main.ts
index d32668187..9f86440d8 100644
--- a/tests/storybook/.storybook/main.ts
+++ b/tests/storybook/.storybook/main.ts
@@ -49,6 +49,11 @@ export default {
           '@toeverything/infra': fileURLToPath(
             new URL('../../../packages/common/infra/src', import.meta.url)
           ),
+          // workaround for https://github.com/vitejs/vite/issues/9731
+          // it seems vite does not resolve self reference correctly?
+          '@affine/core': fileURLToPath(
+            new URL('../../../packages/frontend/core/src', import.meta.url)
+          ),
         },
       },
       plugins: [