From 21cf667b47733c71f3aa1bc2b7f4a591e34ba4e1 Mon Sep 17 00:00:00 2001
From: Korina Cordero <cordero.korina@gmail.com>
Date: Wed, 29 Sep 2021 06:44:04 +0000
Subject: [PATCH] Add check for battery. #624

---
 src/Controller/APIController.php | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/Controller/APIController.php b/src/Controller/APIController.php
index caea00c3..f9f0b596 100644
--- a/src/Controller/APIController.php
+++ b/src/Controller/APIController.php
@@ -2897,10 +2897,14 @@ class APIController extends Controller implements LoggedController
                 }
             }
 
-            // add battery to items
-            $sku = $batt->getSAPCode();
-            if (!empty($sku))  
-                $hub_criteria->addItem($batt->getSAPCode(), 1);
+            // check if batt is null
+            if ($batt != null)
+            {
+                // add battery to items
+                $sku = $batt->getSAPCode();
+                if (!empty($sku))  
+                    $hub_criteria->addItem($batt->getSAPCode(), 1);
+            }
 
             // get customer id. No JO id at this point
             $customer_id = $cust->getID();