diff --git a/models/res_partner_contract.py b/models/res_partner_contract.py
index 1995432..d473e6a 100644
--- a/models/res_partner_contract.py
+++ b/models/res_partner_contract.py
@@ -70,13 +70,10 @@ class PartnerContract(models.Model):
             "context": {"self_id": self.id},
         }
 
-    @api.model
-    def create(self, vals):
-
+    @api.onchange('partner_id')
+    def _onchange_partner_id(self):
         datetime_now = datetime.datetime.now().strftime("%Y-%m-%d")
-        vals["name"] = self._calculate_contract_name(datetime_now)
-
-        return super(PartnerContract, self).create(vals)
+        self.name = self._calculate_contract_name(datetime_now)
 
     def _calculate_contract_name(self, _date):
 
diff --git a/wizard/res_partner_contract_wizard.py b/wizard/res_partner_contract_wizard.py
index 888d139..82b8736 100644
--- a/wizard/res_partner_contract_wizard.py
+++ b/wizard/res_partner_contract_wizard.py
@@ -90,9 +90,8 @@ class ContractWizard(models.TransientModel):
             else self.target.contract_id.partner_id
         )
 
-        self.write(
-            {"company_id": company_id, "partner_id": partner_id,}
-        )
+        self.company_id = company_id
+        self.partner_id = partner_id
 
         model_to_action = {
             "res.partner.contract": "client_contracts.action_get_contract_context",