diff --git a/app/models/spree/payment_decorator.rb b/app/models/spree/payment_decorator.rb
index 8aab22a7bb..6b5f954a10 100644
--- a/app/models/spree/payment_decorator.rb
+++ b/app/models/spree/payment_decorator.rb
@@ -33,7 +33,8 @@ module Spree
     # This is called by the calculator of a payment method
     def line_items
       if order.complete? && Spree::Config[:track_inventory_levels]
-        order.line_items.select { |li| inventory_units.pluck(:variant_id).include?(li.variant_id) }
+        inventory_item_ids = order.distributor.inventory_items.pluck(:variant_id)
+        order.line_items.select { |li| inventory_item_ids.include?(li.variant_id) }
       else
         order.line_items
       end