diff --git a/packages/cli/src/modules/instance-ai/__tests__/compaction.service.test.ts b/packages/cli/src/modules/instance-ai/__tests__/compaction.service.test.ts
index 0b326d7fe16..608d63a0a96 100644
--- a/packages/cli/src/modules/instance-ai/__tests__/compaction.service.test.ts
+++ b/packages/cli/src/modules/instance-ai/__tests__/compaction.service.test.ts
@@ -39,12 +39,9 @@ jest.mock('@n8n/instance-ai', () => ({
 	},
 }));
 
-jest.mock('@mastra/core/agent', () => ({}));
-jest.mock('@mastra/core/storage', () => ({
-	MemoryStorage: class MemoryStorage {},
-	MastraCompositeStore: class MastraCompositeStore {},
+jest.mock('../storage/typeorm-memory-storage', () => ({
+	TypeORMMemoryStorage: class TypeORMMemoryStorage {},
 }));
-jest.mock('@mastra/memory', () => ({}));
 
 import { InstanceAiCompactionService } from '../compaction.service';