diff --git a/account_payment_return_import_iso20022/models/payment_return.py b/account_payment_return_import_iso20022/models/payment_return.py
index 2bbed183b05..4144b5aa42e 100644
--- a/account_payment_return_import_iso20022/models/payment_return.py
+++ b/account_payment_return_import_iso20022/models/payment_return.py
@@ -14,18 +14,17 @@ def _find_match(self):
         for line in self.filtered(lambda x: not x.move_line_ids and x.reference):
             if not line.reference.isdigit():
                 continue
-            payments = self.env["account.payment"].search(
+            payment = self.env["account.payment"].search(
                 [
-                    ("move_id", "=", int(line.reference)),
+                    ("id", "=", int(line.reference)),
                     ("payment_order_id", "!=", False),
                 ],
             )
-            if payments:
-                line.partner_id = payments[0].partner_id
-                for payment in payments:
-                    line.move_line_ids |= payment.move_id.line_ids.filtered(
-                        lambda x, payment=payment: x.account_id
-                        == payment.destination_account_id
-                        and x.partner_id == payment.partner_id
-                    )
+            if payment:
+                line.partner_id = payment.partner_id
+                line.move_line_ids |= payment.move_id.line_ids.filtered(
+                    lambda x, payment=payment: x.account_id
+                    == payment.destination_account_id
+                    and x.partner_id == payment.partner_id
+                )
         return super()._find_match()