diff --git a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
index b30d2d6..c5acab7 100644
--- a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
+++ b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
@@ -89,7 +89,7 @@ class DocumentExtractionService(
                         for ((key, question) in aadharFrontQuestions) {
                             results[key] = llamaClient.ask(localImageUrl, question)
                         }
-                        ensureAadhaarId(localImageUrl, publicImageUrl, results)
+                        ensureAadhaarId(localImageUrl, publicImageUrl, document, results)
                     }
                 }
             ),
@@ -274,7 +274,12 @@ class DocumentExtractionService(
         }
     }
 
-    private fun ensureAadhaarId(localImageUrl: String, publicImageUrl: String, results: MutableMap<String, String>) {
+    private fun ensureAadhaarId(
+        localImageUrl: String,
+        publicImageUrl: String,
+        document: GuestDocument,
+        results: MutableMap<String, String>
+    ) {
         val key = DocumentPrompts.ID_NUMBER.first
         val current = cleanedValue(results[key])
         val normalized = normalizeDigits(current)