diff --git a/account_move_line_sale_info/models/account_move.py b/account_move_line_sale_info/models/account_move.py
index 73ac30030..1c14338a3 100644
--- a/account_move_line_sale_info/models/account_move.py
+++ b/account_move_line_sale_info/models/account_move.py
@@ -12,17 +12,7 @@ class AccountMove(models.Model):
         res = super()._prepare_interim_account_line_vals(
             line, move, debit_interim_account
         )
-        if (
-            not res.get("move_id", False)
-            or not res.get("product_id", False)
-            or not res.get("quantity", False)
-        ):
-            return res
-        am = self.env["account.move"].browse(res["move_id"])
-        sale_line_id = am.invoice_line_ids.filtered(
-            lambda il: il.product_id.id == res["product_id"]
-            and il.quantity == res["quantity"]
-        ).mapped("sale_line_id")
+        sale_line_id = line.sale_line_ids
         if sale_line_id and len(sale_line_id) == 1:
             res["sale_line_id"] = sale_line_id.id
         return res
@@ -31,17 +21,7 @@ class AccountMove(models.Model):
         res = super()._prepare_expense_account_line_vals(
             line, move, debit_interim_account
         )
-        if (
-            not res.get("move_id", False)
-            or not res.get("product_id", False)
-            or not res.get("quantity", False)
-        ):
-            return res
-        am = self.env["account.move"].browse(res["move_id"])
-        sale_line_id = am.invoice_line_ids.filtered(
-            lambda il: il.product_id.id == res["product_id"]
-            and il.quantity == res["quantity"]
-        ).mapped("sale_line_id")
+        sale_line_id = line.sale_line_ids
         if sale_line_id and len(sale_line_id) == 1:
             res["sale_line_id"] = sale_line_id.id
         return res