diff --git a/odex25_realstate/property_management/models/end_rental_contract.py b/odex25_realstate/property_management/models/end_rental_contract.py
index 02a8564b1..f7215346e 100644
--- a/odex25_realstate/property_management/models/end_rental_contract.py
+++ b/odex25_realstate/property_management/models/end_rental_contract.py
@@ -81,6 +81,9 @@ class EndOfRent(models.Model):
     def get_remain_amount(self):
         for rec in self:
             rec.remain_amount = rec.insurance_amount - rec.total_amount
+            
+    def action_draft (self):
+        self.write({'state': 'draft'})
 
     def action_cancel(self):
         if self.state not in ['check', 'done']: