diff --git a/extensions/kilocode/implicit-provider.test.ts b/extensions/kilocode/implicit-provider.test.ts
index b549238d372..67040e3a698 100644
--- a/extensions/kilocode/implicit-provider.test.ts
+++ b/extensions/kilocode/implicit-provider.test.ts
@@ -7,6 +7,6 @@ describe("Kilo Gateway implicit provider", () => {
 
     expect(provider.baseUrl).toBe("https://api.kilo.ai/api/gateway/");
     expect(provider.api).toBe("openai-completions");
-    expect(provider.models?.length).toBeGreaterThan(0);
+    expect(provider.models.length).toBeGreaterThan(0);
   });
 });
diff --git a/extensions/kilocode/provider-models.test.ts b/extensions/kilocode/provider-models.test.ts
index 72c7f663f50..ebb13257638 100644
--- a/extensions/kilocode/provider-models.test.ts
+++ b/extensions/kilocode/provider-models.test.ts
@@ -175,8 +175,8 @@ describe("discoverKilocodeModels (fetch path)", () => {
       expect(sonnet.cost.cacheWrite).toBeCloseTo(3.75);
       expect(sonnet.input).toEqual(["text", "image"]);
       expect(sonnet.reasoning).toBe(true);
-      expect(sonnet?.contextWindow).toBe(200000);
-      expect(sonnet?.maxTokens).toBe(8192);
+      expect(sonnet.contextWindow).toBe(200000);
+      expect(sonnet.maxTokens).toBe(8192);
     });
   });
 
@@ -234,9 +234,9 @@ describe("discoverKilocodeModels (fetch path)", () => {
     });
     await withFetchPathTest(mockFetch, async () => {
       const models = await discoverKilocodeModels();
-      const textModel = models.find((m) => m.id === "some/text-model");
-      expect(textModel?.input).toEqual(["text"]);
-      expect(textModel?.reasoning).toBe(false);
+      const textModel = requireModelById(models, "some/text-model");
+      expect(textModel.input).toEqual(["text"]);
+      expect(textModel.reasoning).toBe(false);
     });
   });