diff --git a/react-vite/base/vite.config.ts b/react-vite/base/vite.config.ts
index ba7e47f5c..89f5ecd22 100644
--- a/react-vite/base/vite.config.ts
+++ b/react-vite/base/vite.config.ts
@@ -1,3 +1,5 @@
+/// <reference types="vitest" />
+
 import legacy from '@vitejs/plugin-legacy'
 import react from '@vitejs/plugin-react'
 import { defineConfig } from 'vite'
diff --git a/vue-vite/base/vite.config.ts b/vue-vite/base/vite.config.ts
index 5f24224fd..07c35d0d2 100644
--- a/vue-vite/base/vite.config.ts
+++ b/vue-vite/base/vite.config.ts
@@ -1,3 +1,5 @@
+/// <reference types="vitest" />
+
 import legacy from '@vitejs/plugin-legacy'
 import vue from '@vitejs/plugin-vue'
 import path from 'path'