diff --git a/odex25_project/project_base/models/project_invoice.py b/odex25_project/project_base/models/project_invoice.py
index dd8f83466..ab36044d3 100644
--- a/odex25_project/project_base/models/project_invoice.py
+++ b/odex25_project/project_base/models/project_invoice.py
@@ -233,14 +233,14 @@ class ProjectInvoice(models.Model):
 
     def action_set_to_draft(self):
         self.ensure_one()
-        if self.invoice_id and self.invoice_id.state=='draft':
-            self.invoice_id.sudo().write({'posted_before':False})
-            self.invoice_id.sudo().unlink()
-            self.state = 'draft'
-        else:
-            if self.invoice_id:
+        if self.invoice_id and self.invoice_id.state!='draft':
                 raise UserError(_("Kindly The invoice is not in draft state, so it cannot be unlinked."))
 
+        if self.invoice_id:
+            self.invoice_id.sudo().write({'posted_before':False})
+            self.invoice_id.sudo().unlink()
+        self.state = 'draft'
+        
     def action_get_invoice(self):
         self.ensure_one()
         view_id = False