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>