diff --git a/extensions/openrouter/index.test.ts b/extensions/openrouter/index.test.ts
index ff633d3f9d5..51690e84d25 100644
--- a/extensions/openrouter/index.test.ts
+++ b/extensions/openrouter/index.test.ts
@@ -85,7 +85,9 @@ describe("openrouter provider hooks", () => {
     );
 
     expect(baseStreamFn).toHaveBeenCalledOnce();
-    expect(baseStreamFn.mock.calls[0]?.[0]).toMatchObject({
+    const firstCall = baseStreamFn.mock.calls[0];
+    const firstModel = firstCall?.[0];
+    expect(firstModel).toMatchObject({
       compat: {
         openRouterRouting: {
           order: ["moonshot"],
diff --git a/extensions/xai/stream.test.ts b/extensions/xai/stream.test.ts
index c73ef03cd1f..4787442a2b1 100644
--- a/extensions/xai/stream.test.ts
+++ b/extensions/xai/stream.test.ts
@@ -10,7 +10,10 @@ import {
 type ToolPayload = {
   function?: Record<string, unknown>;
 };
-
+type XaiTestPayload = Record<string, unknown> & {
+  tools?: Array<{ type?: string; function?: Record<string, unknown> }>;
+  input?: unknown[];
+};
 function captureWrappedModelId(params: {
   modelId: string;
   fastMode: boolean;
@@ -63,10 +66,10 @@ describe("xai stream wrappers", () => {
 
   it("composes the xai provider stream chain from extra params", () => {
     let capturedModelId = "";
-    let capturedPayload: Record<string, unknown> | undefined;
+    let capturedPayload: XaiTestPayload | undefined;
     const baseStreamFn: StreamFn = (model, _context, options) => {
       capturedModelId = String(model.id);
-      const payload: Record<string, unknown> = {
+      const payload: XaiTestPayload = {
         reasoning: { effort: "high" },
         tools: [
           {