diff --git a/src/Service/InvoiceCreator.php b/src/Service/InvoiceCreator.php
index 500a5ad0..df33c0e2 100644
--- a/src/Service/InvoiceCreator.php
+++ b/src/Service/InvoiceCreator.php
@@ -206,9 +206,10 @@ class InvoiceCreator
     public function processWarranty(&$total, InvoiceCriteria $criteria, $invoice)
     {
         error_log('processing warranty');
-        $batteries = $criteria->getBatteries();
-        foreach ($batteries as $batt)
+        $entries = $criteria->getEntries();
+        foreach ($entries as $entry)
         {
+            $batt = $entry['battery'];
             $item = new InvoiceItem();
             $item->setInvoice($invoice)
                 ->setTitle($batt->getModel()->getName() . ' ' . $batt->getSize()->getName() . ' - Service Unit')