diff --git a/modules/purchase_trade/service.py b/modules/purchase_trade/service.py
index 6bfa9e3..927fa8d 100644
--- a/modules/purchase_trade/service.py
+++ b/modules/purchase_trade/service.py
@@ -56,7 +56,8 @@ class ContractFactory:
             contract.reference = c.reference
             contract.from_location = c.from_location
             contract.to_location = c.to_location
-            contract.company = 6
+            context = Transaction().context
+            contract.company = context.get('company') if context else None
             if type_ == 'Purchase':
                 contract.purchase_date = Date.today()
             else: