diff --git a/account_payment_group/models/account_move.py b/account_payment_group/models/account_move.py
index 81080b625..c5255c246 100644
--- a/account_payment_group/models/account_move.py
+++ b/account_payment_group/models/account_move.py
@@ -89,6 +89,14 @@ def action_post(self):
         self.pay_now()
         return res
 
+    def pay_now_context(self, partner_type):
+        self.ensure_one()
+        return {
+            'to_pay_move_line_ids': (self.open_move_line_ids.ids),
+            'default_company_id': self.company_id.id,
+            'default_partner_type': partner_type,
+        }
+
     def pay_now(self):
         # validate_payment = not self._context.get('validate_payment')
         for rec in self:
@@ -104,11 +112,7 @@ def pay_now(self):
                 else:
                     partner_type = 'customer'
 
-                pay_context = {
-                    'to_pay_move_line_ids': (rec.open_move_line_ids.ids),
-                    'default_company_id': rec.company_id.id,
-                    'default_partner_type': partner_type,
-                }
+                pay_context = rec.pay_now_context(partner_type)
 
                 payment_group = rec.env[
                     'account.payment.group'].with_context(