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>