diff --git a/netforce_clinic/models/shop.py b/netforce_clinic/models/shop.py
index c664886..143a3d1 100644
--- a/netforce_clinic/models/shop.py
+++ b/netforce_clinic/models/shop.py
@@ -444,7 +444,6 @@ class Shop(Model):
             for payment in obj.payments:
                 payment.to_draft()
                 payment.delete()
-
             for inv in obj.invoices:
                 inv.write({
                     'state': 'draft',
@@ -453,11 +452,9 @@ class Shop(Model):
                     inv.move_id.to_draft()
                     inv.move_id.delete()
                 inv.delete()
-
             for picking in obj.pickings:
-                picking.to_draft()
+                picking.to_draft(context=context)
                 picking.delete()
-
             obj.write({
                 'state': 'draft',
             })