diff --git a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
index f74986f09565691f93a8b961c146c065fdc834bd..e96124d67a453418aae239c286abc2b01f791ae1 100644
--- a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
+++ b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
@@ -146,11 +146,12 @@ result = line.supply_point_assignation_id.distribution_table_id.selfconsumption_
             self._prepare_contract_values(journal_id, contract_line)
         )
 
+        product_id.write({"contract_template_id": contract_template_id.id})
+
         self.selfconsumption_id.write(
             {
                 "invoicing_mode": self.invoicing_mode,
                 "product_id": product_id,
-                "contract_template_id": contract_template_id,
             }
         )