diff --git a/test/cable_unittest.cpp b/test/cable_unittest.cpp
index b181584..39384f8 100644
--- a/test/cable_unittest.cpp
+++ b/test/cable_unittest.cpp
@@ -1,6 +1,6 @@
 #include "cable.hpp"
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/cpld_unittest.cpp b/test/cpld_unittest.cpp
index 12a0444..8b22ff1 100644
--- a/test/cpld_unittest.cpp
+++ b/test/cpld_unittest.cpp
@@ -1,6 +1,6 @@
 #include "cpld.hpp"
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 
 #include <cstdint>
 #include <tuple>
diff --git a/test/entity_unittest.cpp b/test/entity_unittest.cpp
index 0a76d86..5ef6d82 100644
--- a/test/entity_unittest.cpp
+++ b/test/entity_unittest.cpp
@@ -1,6 +1,6 @@
 #include "entity_name.hpp"
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/eth_unittest.cpp b/test/eth_unittest.cpp
index af1a9cc..6c4b98a 100644
--- a/test/eth_unittest.cpp
+++ b/test/eth_unittest.cpp
@@ -1,6 +1,6 @@
 #include "eth.hpp"
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 
 #include <cstdint>
 #include <cstring>
diff --git a/test/pcie_unittest.cpp b/test/pcie_unittest.cpp
index 84bbe5c..796d97a 100644
--- a/test/pcie_unittest.cpp
+++ b/test/pcie_unittest.cpp
@@ -1,5 +1,5 @@
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 #include "pcie_i2c.hpp"
 
 #include <cstdint>
diff --git a/test/psu_unittest.cpp b/test/psu_unittest.cpp
index 7805370..c2a7a53 100644
--- a/test/psu_unittest.cpp
+++ b/test/psu_unittest.cpp
@@ -1,5 +1,5 @@
 #include "handler_mock.hpp"
-#include "main.hpp"
+#include "ipmi.hpp"
 #include "psu.hpp"
 
 #include <cstdint>