diff --git a/product_contract/models/sale_order.py b/product_contract/models/sale_order.py
index 6eb6a973..0d528a51 100644
--- a/product_contract/models/sale_order.py
+++ b/product_contract/models/sale_order.py
@@ -19,7 +19,7 @@ class SaleOrder(models.Model):
     @api.depends('order_line.contract_id', 'state')
     def _compute_need_contract_creation(self):
         for rec in self:
-            if rec.state not in ('draft', 'sent'):
+            if rec.state in ('sale', 'done'):
                 line_to_create_contract = rec.order_line.filtered(
                     lambda r: not r.contract_id and r.product_id.is_contract
                 )