diff --git a/packages/cli/src/public-ingest.test.ts b/packages/cli/src/public-ingest.test.ts
index 5d714876..d2357f23 100644
--- a/packages/cli/src/public-ingest.test.ts
+++ b/packages/cli/src/public-ingest.test.ts
@@ -417,7 +417,7 @@ describe('runKtxPublicIngest', () => {
       },
     });
     const runScan = vi.fn(async () => 0);
-    const runIngest = vi.fn(async () => 0);
+    const runIngest = vi.fn<NonNullable<KtxPublicIngestDeps['runIngest']>>(async () => 0);
 
     await expect(
       runKtxPublicIngest(
@@ -436,7 +436,9 @@ describe('runKtxPublicIngest', () => {
       ),
     ).resolves.toBe(0);
 
-    const ingestArgs = runIngest.mock.calls[0]?.[0];
+    const ingestArgs = runIngest.mock.calls[0]?.[0] as
+      | Extract<Parameters<NonNullable<KtxPublicIngestDeps['runIngest']>>[0], { command: 'run' }>
+      | undefined;
     expect(ingestArgs).toMatchObject({
       command: 'run',
       connectionId: 'warehouse',