diff --git a/sale.py b/sale.py
index 1fd789c..e09d473 100644
--- a/sale.py
+++ b/sale.py
@@ -33,9 +33,10 @@ class Sale(metaclass=PoolMeta):
         Party = Pool().get('party.party')
         User = Pool().get('res.user')
         ctx = Transaction().context
-        print('args.....', args)
         if ctx.get('shop'):
             shop = Shop(ctx['shop'])
+        if args.get('shop'):
+            shop = Shop(args['shop'])
         else:
             user_id = ctx.get('user')
             user = User(user_id)