diff --git a/test/controller_mock.hpp b/test/controller_mock.hpp
index d2375b1..9a8e7a6 100644
--- a/test/controller_mock.hpp
+++ b/test/controller_mock.hpp
@@ -1,6 +1,7 @@
 #pragma once
 
 #include "pid/controller.hpp"
+#include "pid/pidcontroller.hpp"
 #include "pid/zone_interface.hpp"
 
 #include <gmock/gmock.h>
diff --git a/test/dbushelper_mock.hpp b/test/dbushelper_mock.hpp
index 0d4221d..cdbc59d 100644
--- a/test/dbushelper_mock.hpp
+++ b/test/dbushelper_mock.hpp
@@ -1,11 +1,11 @@
 #pragma once
 
+#include "dbus/dbushelper_interface.hpp"
 #include "util.hpp"
 
 #include <string>
 
 #include <gmock/gmock.h>
-
 namespace pid_control
 {