diff --git a/models/__pycache__/account_move_line.cpython-312.pyc b/models/__pycache__/account_move_line.cpython-312.pyc
index e1c1164..d044f39 100644
Binary files a/models/__pycache__/account_move_line.cpython-312.pyc and b/models/__pycache__/account_move_line.cpython-312.pyc differ
diff --git a/models/account_move_line.py b/models/account_move_line.py
index 6570e15..55ef4f9 100755
--- a/models/account_move_line.py
+++ b/models/account_move_line.py
@@ -144,10 +144,19 @@ class AccountMoveLine(models.Model):
 
                     # Update PO Line
                     # We update both to ensure consistency
+                    
+                    po = po_line.order_id
+                    was_locked = po.state == 'done'
+                    if was_locked:
+                        po.button_unlock()
+
                     po_line.write({
                         'price_unit': price_unit,
                         'product_qty': product_qty,
                     })
+
+                    if was_locked:
+                        po.button_done()
                         
         return res