diff --git a/enterprise/app/services/captain/roleta/offer_service.rb b/enterprise/app/services/captain/roleta/offer_service.rb
index 852f5be19..052afcaf2 100644
--- a/enterprise/app/services/captain/roleta/offer_service.rb
+++ b/enterprise/app/services/captain/roleta/offer_service.rb
@@ -48,7 +48,7 @@ class Captain::Roleta::OfferService
   end
 
   def fetch_unidade_for_conversation(conversation)
-    unit = conversation&.inbox&.captain_inbox&.unit
+    unit = conversation&.inbox&.captain_inbox&.captain_unit
     return nil if unit.blank?
 
     supabase_get('unidades', { chatwoot_unit_id: "eq.#{unit.id}", select: '*', limit: 1 }).first
diff --git a/enterprise/app/services/captain/tools/create_reservation_intent_tool.rb b/enterprise/app/services/captain/tools/create_reservation_intent_tool.rb
index 345f8b16b..e72346c99 100644
--- a/enterprise/app/services/captain/tools/create_reservation_intent_tool.rb
+++ b/enterprise/app/services/captain/tools/create_reservation_intent_tool.rb
@@ -256,7 +256,7 @@ class Captain::Tools::CreateReservationIntentTool < Captain::Tools::BaseTool
   end
 
   def infer_unit
-    @conversation&.inbox&.captain_inbox&.unit
+    @conversation&.inbox&.captain_inbox&.captain_unit
   end
 
   def update_sticky_state(suite:, price:, check_in_at:, check_out_at:)
diff --git a/enterprise/app/services/captain/tools/get_reserva_preco_tool.rb b/enterprise/app/services/captain/tools/get_reserva_preco_tool.rb
index d449a43ca..fae848229 100644
--- a/enterprise/app/services/captain/tools/get_reserva_preco_tool.rb
+++ b/enterprise/app/services/captain/tools/get_reserva_preco_tool.rb
@@ -76,7 +76,7 @@ class Captain::Tools::GetReservaPrecoTool < Captain::Tools::BaseTool
   private
 
   def infer_unit
-    @conversation&.inbox&.captain_inbox&.unit
+    @conversation&.inbox&.captain_inbox&.captain_unit
   end
 
   def fetch_unidade(chatwoot_unit_id)