diff --git a/main/vitest.config.ts b/main/vitest.config.ts
index 7794efae..6e7ece4d 100644
--- a/main/vitest.config.ts
+++ b/main/vitest.config.ts
@@ -35,6 +35,7 @@ export default defineConfig({
   define: {
     __APP_VERSION__: `"${Package.version}"`,
     __APP_PLATFORM__: '"chrome"',
+    __EXTENSION_CONTEXT__: '"page"',
   },
   assetsInclude: [
     "**/*.wasm",