diff --git a/account_payment_return_import_iso20022/models/payment_return.py b/account_payment_return_import_iso20022/models/payment_return.py
index 2bbed183b05..8506c823f39 100644
--- a/account_payment_return_import_iso20022/models/payment_return.py
+++ b/account_payment_return_import_iso20022/models/payment_return.py
@@ -16,10 +16,17 @@ def _find_match(self):
                 continue
             payments = self.env["account.payment"].search(
                 [
-                    ("move_id", "=", int(line.reference)),
+                    ("id", "=", int(line.reference)),
                     ("payment_order_id", "!=", False),
                 ],
             )
+            if not payments:
+                payments = self.env["account.payment"].search(
+                    [
+                        ("move_id", "=", int(line.reference)),
+                        ("payment_order_id", "!=", False),
+                    ],
+                )
             if payments:
                 line.partner_id = payments[0].partner_id
                 for payment in payments: