diff --git a/src/Controller/CustomerAppAPI/InvoiceController.php b/src/Controller/CustomerAppAPI/InvoiceController.php
index c4bc132f..67522c96 100644
--- a/src/Controller/CustomerAppAPI/InvoiceController.php
+++ b/src/Controller/CustomerAppAPI/InvoiceController.php
@@ -118,8 +118,9 @@ class InvoiceController extends ApiController
 
         // if we have a trade in, add it as well
         if (!empty($trade_in_type) && !empty($trade_in_batt)) {
-            $ti_batt_size_obj = $this->em->getRepository(BatterySize::class)->find($trade_in_batt);
-            if (!empty($ti_batt_size_obj)) {
+            $ti_batt_obj = $this->em->getRepository(Battery::class)->find($trade_in_batt);
+            if (!empty($ti_batt_obj)) {
+                $ti_batt_size_obj = $ti_batt_obj->getSize();
                 $icrit->addTradeInEntry($ti_batt_size_obj, $trade_in_type, 1);
             }
         }
diff --git a/src/Controller/CustomerAppAPI/JobOrderController.php b/src/Controller/CustomerAppAPI/JobOrderController.php
index 0082de25..839cf226 100644
--- a/src/Controller/CustomerAppAPI/JobOrderController.php
+++ b/src/Controller/CustomerAppAPI/JobOrderController.php
@@ -689,8 +689,9 @@ class JobOrderController extends ApiController
 
         // if we have a trade in, add it as well
         if (!empty($trade_in_type) && !empty($trade_in_batt)) {
-            $ti_batt_size_obj = $this->em->getRepository(BatterySize::class)->find($trade_in_batt);
-            if (!empty($ti_batt_size_obj)) {
+            $ti_batt_obj = $this->em->getRepository(Battery::class)->find($trade_in_batt);
+            if (!empty($ti_batt_obj)) {
+                $ti_batt_size_obj = $ti_batt_obj->getSize();
                 $icrit->addTradeInEntry($ti_batt_size_obj, $trade_in_type, 1);
             }
         }