diff --git a/test/zone_mock.hpp b/test/zone_mock.hpp
index dcf6b2a..7d3a199 100644
--- a/test/zone_mock.hpp
+++ b/test/zone_mock.hpp
@@ -1,8 +1,14 @@
 #pragma once
 
+#include "interfaces.hpp"
 #include "pid/zone_interface.hpp"
+#include "sensors/sensor.hpp"
 
+#include <cstdint>
+#include <map>
 #include <string>
+#include <string_view>
+#include <utility>
 #include <vector>
 
 #include <gmock/gmock.h>