1 files changed, 4 insertions, 0 deletions
diff --git a/packages/core/src/utils/memoryDiscovery.test.ts b/packages/core/src/utils/memoryDiscovery.test.ts
index db0ffd1d..a9d34bf3 100644
--- a/packages/core/src/utils/memoryDiscovery.test.ts
+++ b/packages/core/src/utils/memoryDiscovery.test.ts
@@ -45,6 +45,10 @@ describe('loadServerHierarchicalMemory', () => {
 
   beforeEach(() => {
     vi.resetAllMocks();
+    // Set environment variables to indicate test environment
+    process.env.NODE_ENV = 'test';
+    process.env.VITEST = 'true';
+
     setGeminiMdFilename(DEFAULT_CONTEXT_FILENAME); // Use defined const
     mockOs.homedir.mockReturnValue(USER_HOME);