diff --git a/packages/cli/src/ui/components/Footer.test.tsx b/packages/cli/src/ui/components/Footer.test.tsx
index 6ff46ef7b..f2b759e69 100644
--- a/packages/cli/src/ui/components/Footer.test.tsx
+++ b/packages/cli/src/ui/components/Footer.test.tsx
@@ -23,7 +23,7 @@ const defaultProps = {
 const createMockConfig = (overrides = {}) => ({
   getModel: vi.fn(() => defaultProps.model),
   getDebugMode: vi.fn(() => false),
-  getContentGeneratorConfig: vi.fn(() => ({})),
+  getContentGeneratorConfig: vi.fn(() => ({ contextWindowSize: 131072 })),
   getMcpServers: vi.fn(() => ({})),
   getBlockedMcpServers: vi.fn(() => []),
   ...overrides,
diff --git a/packages/core/src/core/tokenLimits.test.ts b/packages/core/src/core/tokenLimits.test.ts
index b2cbbd246..ffd71cd4b 100644
--- a/packages/core/src/core/tokenLimits.test.ts
+++ b/packages/core/src/core/tokenLimits.test.ts
@@ -235,20 +235,13 @@ describe('tokenLimit', () => {
   });
 
   describe('Moonshot Kimi', () => {
-    it('should return the correct limit for kimi-k2-0905-preview', () => {
+    it('should return the correct limit for kimi-k2 variants', () => {
       expect(tokenLimit('kimi-k2-0905-preview')).toBe(262144); // 256K
       expect(tokenLimit('kimi-k2-0905')).toBe(262144);
-    });
-    it('should return the correct limit for kimi-k2-turbo-preview', () => {
-      expect(tokenLimit('kimi-k2-turbo-preview')).toBe(262144); // 256K
+      expect(tokenLimit('kimi-k2-turbo-preview')).toBe(262144);
       expect(tokenLimit('kimi-k2-turbo')).toBe(262144);
-    });
-    it('should return the correct limit for kimi-k2-0711-preview', () => {
-      expect(tokenLimit('kimi-k2-0711-preview')).toBe(131072); // 128K
-      expect(tokenLimit('kimi-k2-0711')).toBe(131072);
-    });
-    it('should return the correct limit for kimi-k2-instruct', () => {
-      expect(tokenLimit('kimi-k2-instruct')).toBe(131072); // 128K
+      expect(tokenLimit('kimi-k2-0711-preview')).toBe(262144);
+      expect(tokenLimit('kimi-k2-instruct')).toBe(262144);
     });
   });