diff --git a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
index 1888581..3048a28 100644
--- a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
+++ b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
@@ -427,7 +427,8 @@ class DocumentExtractionService(
     }
 
     private fun applyBookingCityUpdates(document: GuestDocument, results: Map<String, String>) {
-        val booking = document.booking
+        val bookingId = document.booking?.id ?: return
+        val booking = bookingRepo.findById(bookingId).orElse(null) ?: return
         if (booking.fromCity?.isNotBlank() == true && booking.toCity?.isNotBlank() == true) return
         val pin = cleanedValue(results[DocumentPrompts.PIN_CODE.first]) ?: return
         if (!isValidPin(pin)) return