diff --git a/vite.config.ts b/vite.config.ts
index e2b0e15c0..8d34e6184 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -9,7 +9,8 @@ export default defineConfig({
   test: {
     environment: 'jsdom',
     globals: true,
-    setupFiles: ['src/tests/setup.ts'],
+    // TODO: should we move this to a mermaid-core package?
+    setupFiles: ['packages/mermaid/src/tests/setup.ts'],
     coverage: {
       reporter: ['text', 'json', 'html', 'lcov'],
     },