diff --git a/main.cpp b/main.cpp
index eaa82ab..f0c61a9 100644
--- a/main.cpp
+++ b/main.cpp
@@ -19,6 +19,7 @@
 #include "cable.hpp"
 #include "cpld.hpp"
 #include "eth.hpp"
+#include "pcie_i2c.hpp"
 #include "psu.hpp"
 
 #include <host-ipmid/ipmid-api.h>
@@ -62,6 +63,10 @@
             return GetEthDevice(reqBuf, replyCmdBuf, dataLen);
         case SysPsuHardReset:
             return PsuHardReset(reqBuf, replyCmdBuf, dataLen);
+        case SysPcieSlotCount:
+            return PcieSlotCount(reqBuf, replyCmdBuf, dataLen);
+        case SysPcieSlotI2cBusMapping:
+            return PcieSlotI2cBusMapping(reqBuf, replyCmdBuf, dataLen);
         default:
             std::fprintf(stderr, "Invalid subcommand: 0x%x\n", reqBuf[0]);
             return IPMI_CC_INVALID;