diff --git a/nvme_manager.hpp b/nvme_manager.hpp
index 78eabe3..391a768 100644
--- a/nvme_manager.hpp
+++ b/nvme_manager.hpp
@@ -119,7 +119,8 @@
 
     /** @brief Set inventory properties of nvme */
     void setNvmeInventoryProperties(
-        bool present, const phosphor::nvme::Nvme::NVMeData& nvmeData,
+        NVMeConfig& config, bool present,
+        const phosphor::nvme::Nvme::NVMeData& nvmeData,
         const std::string& inventoryPath);
 
     void createNVMeInventory();