diff --git a/test/cable_unittest.cpp b/test/cable_unittest.cpp
index 39384f8..2eeb486 100644
--- a/test/cable_unittest.cpp
+++ b/test/cable_unittest.cpp
@@ -1,6 +1,6 @@
 #include "cable.hpp"
+#include "commands.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/cpld_unittest.cpp b/test/cpld_unittest.cpp
index 8b22ff1..2054df7 100644
--- a/test/cpld_unittest.cpp
+++ b/test/cpld_unittest.cpp
@@ -1,6 +1,6 @@
+#include "commands.hpp"
 #include "cpld.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 
 #include <cstdint>
 #include <tuple>
diff --git a/test/entity_unittest.cpp b/test/entity_unittest.cpp
index 5ef6d82..7091cb1 100644
--- a/test/entity_unittest.cpp
+++ b/test/entity_unittest.cpp
@@ -1,6 +1,6 @@
+#include "commands.hpp"
 #include "entity_name.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/eth_unittest.cpp b/test/eth_unittest.cpp
index 6c4b98a..1053d4b 100644
--- a/test/eth_unittest.cpp
+++ b/test/eth_unittest.cpp
@@ -1,6 +1,6 @@
+#include "commands.hpp"
 #include "eth.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/pcie_unittest.cpp b/test/pcie_unittest.cpp
index 796d97a..5f7130f 100644
--- a/test/pcie_unittest.cpp
+++ b/test/pcie_unittest.cpp
@@ -1,5 +1,5 @@
+#include "commands.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 #include "pcie_i2c.hpp"
 
 #include <cstdint>
diff --git a/test/psu_unittest.cpp b/test/psu_unittest.cpp
index c2a7a53..7127c3d 100644
--- a/test/psu_unittest.cpp
+++ b/test/psu_unittest.cpp
@@ -1,5 +1,5 @@
+#include "commands.hpp"
 #include "handler_mock.hpp"
-#include "ipmi.hpp"
 #include "psu.hpp"
 
 #include <cstdint>