diff --git a/apps/api/src/lib/extract/extraction-service.ts b/apps/api/src/lib/extract/extraction-service.ts
index 6f9e99a8..0ac30245 100644
--- a/apps/api/src/lib/extract/extraction-service.ts
+++ b/apps/api/src/lib/extract/extraction-service.ts
@@ -80,7 +80,7 @@ async function analyzeSchemaAndPrompt(
     multiEntityKeys: z.array(z.string()).optional().default([]),
     reasoning: z.string(),
     keyIndicators: z.array(z.string()),
-  }).refine(x => !x.isMultiEntity || multiEntityKeys.length > 0, "isMultiEntity was true, but no multiEntityKeys were returned");
+  }).refine(x => !x.isMultiEntity || x.multiEntityKeys.length > 0, "isMultiEntity was true, but no multiEntityKeys");
 
   const model = "gpt-4o";