diff --git a/Model/Write/Products/CollectionDecorator/StockData/SourceItemMapProvider.php b/Model/Write/Products/CollectionDecorator/StockData/SourceItemMapProvider.php
index 6c91ad6..60451b8 100644
--- a/Model/Write/Products/CollectionDecorator/StockData/SourceItemMapProvider.php
+++ b/Model/Write/Products/CollectionDecorator/StockData/SourceItemMapProvider.php
@@ -111,7 +111,7 @@ public function getStockItemMap(Collection|StockCollection $collection): array
             return [];
         }
 
-        $skus = $collection->getAllSkus();
+        $entityIds = $collection->getAllIds();
 
         $store = $collection->getStore();
         $sourceCodes = $this->getSourceCodesForStore($store);
@@ -159,7 +159,7 @@ public function getStockItemMap(Collection|StockCollection $collection): array
                 ->select()
                 ->from($stockItemTable)
                 ->reset('columns')
-                ->where("$stockItemTable.product_id IN (?)", $collection->getAllIds())
+                ->where("$stockItemTable.product_id IN (?)", $entityIds)
                 /*
                 $stock_id is in this case the default stock id (i.e. 1) this filter problably doesnt remove anything
                 but it is here just to be sure.
@@ -207,7 +207,7 @@ public function getStockItemMap(Collection|StockCollection $collection): array
                 'backorders',
             ]
         )
-        ->where("$productTableName.sku IN (?)", $skus)
+        ->where("$productTableName.entity_id IN (?)", $entityIds)
         ->columns(
             [
                 'product_entity_id' => "$productTableName.entity_id",