diff --git a/packages/data-schemas/src/methods/conversation.spec.ts b/packages/data-schemas/src/methods/conversation.spec.ts
index 1e17092067..bfdd55e832 100644
--- a/packages/data-schemas/src/methods/conversation.spec.ts
+++ b/packages/data-schemas/src/methods/conversation.spec.ts
@@ -1,6 +1,6 @@
 import mongoose from 'mongoose';
 import { v4 as uuidv4 } from 'uuid';
-import { EModelEndpoint } from 'librechat-data-provider';
+import { EModelEndpoint, RetentionMode } from 'librechat-data-provider';
 import type { IConversation } from '../types';
 import { MongoMemoryServer } from 'mongodb-memory-server';
 import { ConversationMethods, createConversationMethods } from './conversation';
@@ -78,7 +78,7 @@ describe('Conversation Operations', () => {
   let mockCtx: {
     userId: string;
     isTemporary?: boolean;
-    interfaceConfig?: { temporaryChatRetention?: number };
+    interfaceConfig?: { temporaryChatRetention?: number; retentionMode?: RetentionMode };
   };
   let mockConversationData: {
     conversationId: string;
@@ -402,7 +402,7 @@ describe('Conversation Operations', () => {
       mockCtx.isTemporary = false;
       mockCtx.interfaceConfig = {
         temporaryChatRetention: 24,
-        retentionMode: 'all',
+        retentionMode: RetentionMode.ALL,
       };
       const result = await saveConvo(mockCtx, mockConversationData);
       expect(result?.expiredAt).toBeDefined();
@@ -413,7 +413,7 @@ describe('Conversation Operations', () => {
       mockCtx.isTemporary = false;
       mockCtx.interfaceConfig = {
         temporaryChatRetention: 24,
-        retentionMode: 'temporary',
+        retentionMode: RetentionMode.TEMPORARY,
       };
       const result = await saveConvo(mockCtx, mockConversationData);
       expect(result?.expiredAt).toBeNull();
diff --git a/packages/data-schemas/src/methods/message.spec.ts b/packages/data-schemas/src/methods/message.spec.ts
index ce7ebafdf9..440ae6003d 100644
--- a/packages/data-schemas/src/methods/message.spec.ts
+++ b/packages/data-schemas/src/methods/message.spec.ts
@@ -1,5 +1,6 @@
 import mongoose from 'mongoose';
 import { v4 as uuidv4 } from 'uuid';
+import { RetentionMode } from 'librechat-data-provider';
 import { MongoMemoryServer } from 'mongodb-memory-server';
 import type { IMessage } from '..';
 import { createMessageMethods } from './message';
@@ -54,7 +55,7 @@ describe('Message Operations', () => {
   let mockCtx: {
     userId: string;
     isTemporary?: boolean;
-    interfaceConfig?: { temporaryChatRetention?: number };
+    interfaceConfig?: { temporaryChatRetention?: number; retentionMode?: RetentionMode };
   };
   let mockMessageData: Partial<IMessage> = {
     messageId: 'msg123',
@@ -479,7 +480,7 @@ describe('Message Operations', () => {
       mockCtx.isTemporary = false;
       mockCtx.interfaceConfig = {
         temporaryChatRetention: 24,
-        retentionMode: 'all',
+        retentionMode: RetentionMode.ALL,
       };
       const result = await saveMessage(mockCtx, mockMessageData);
       expect(result?.expiredAt).toBeDefined();
@@ -490,7 +491,7 @@ describe('Message Operations', () => {
       mockCtx.isTemporary = false;
       mockCtx.interfaceConfig = {
         temporaryChatRetention: 24,
-        retentionMode: 'temporary',
+        retentionMode: RetentionMode.TEMPORARY,
       };
       const result = await saveMessage(mockCtx, mockMessageData);
       expect(result?.expiredAt).toBeNull();