diff --git a/test/openpower-pels/mocks.hpp b/test/openpower-pels/mocks.hpp
index e291966..0e55ecf 100644
--- a/test/openpower-pels/mocks.hpp
+++ b/test/openpower-pels/mocks.hpp
@@ -39,7 +39,7 @@
     MOCK_METHOD(std::string, expandLocationCode, (const std::string&, uint16_t),
                 (const override));
     MOCK_METHOD(std::string, getInventoryFromLocCode,
-                (const std::string&, uint16_t), (const override));
+                (const std::string&, uint16_t, bool), (const override));
 
     void changeHostState(bool newState)
     {
diff --git a/test/openpower-pels/pel_test.cpp b/test/openpower-pels/pel_test.cpp
index d643076..b7047cc 100644
--- a/test/openpower-pels/pel_test.cpp
+++ b/test/openpower-pels/pel_test.cpp
@@ -817,7 +817,7 @@
                     "/xyz/openbmc_project/inventory/chassis/motherboard/cpu0"))
         .WillOnce(Return("UXXX-P1"));
 
-    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1", 0))
+    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1", 0, false))
         .WillOnce(
             Return("/xyz/openbmc_project/inventory/chassis/motherboard/cpu0"));
 
diff --git a/test/openpower-pels/src_test.cpp b/test/openpower-pels/src_test.cpp
index cac653a..c4c9d40 100644
--- a/test/openpower-pels/src_test.cpp
+++ b/test/openpower-pels/src_test.cpp
@@ -618,11 +618,11 @@
         EXPECT_CALL(dataIface, expandLocationCode("P0-C9", 0))
             .WillOnce(Return("UXXX-P0-C9"));
 
-        EXPECT_CALL(dataIface, getInventoryFromLocCode("P0-C8", 0))
+        EXPECT_CALL(dataIface, getInventoryFromLocCode("P0-C8", 0, false))
             .WillOnce(Return(
                 "/xyz/openbmc_project/inventory/chassis/motherboard/cpu0"));
 
-        EXPECT_CALL(dataIface, getInventoryFromLocCode("P0-C9", 0))
+        EXPECT_CALL(dataIface, getInventoryFromLocCode("P0-C9", 0, false))
             .WillOnce(Return(
                 "/xyz/openbmc_project/inventory/chassis/motherboard/cpu1"));
 
@@ -721,17 +721,17 @@
         .Times(5)
         .WillRepeatedly(ReturnRef(names));
 
-    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1-C40", 0))
+    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1-C40", 0, false))
         .Times(3)
         .WillRepeatedly(
             Return("/xyz/openbmc_project/inventory/chassis/motherboard/cpu0"));
 
-    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1", 0))
+    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1", 0, false))
         .Times(3)
         .WillRepeatedly(
             Return("/xyz/openbmc_project/inventory/chassis/motherboard"));
 
-    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1-C15", 0))
+    EXPECT_CALL(dataIface, getInventoryFromLocCode("P1-C15", 0, false))
         .Times(3)
         .WillRepeatedly(
             Return("/xyz/openbmc_project/inventory/chassis/motherboard/bmc"));