diff --git a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
index 6bcdcedae2371de1e7389c0296ad9ae5b13a2b57..93bbe2a1a0e4524ed9b1e169996ec14d781eed67 100644
--- a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
+++ b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
@@ -160,11 +160,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,
             }
         )