From c244cf658cff66bcda7322d06693b7ef4e411a14 Mon Sep 17 00:00:00 2001
From: laurentbarontini <l.barontini@open-squared.ch>
Date: Mon, 23 Feb 2026 15:33:09 +0100
Subject: [PATCH] 23.02.26

---
 modules/purchase_trade/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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: