diff --git a/test/controller_mock.hpp b/test/controller_mock.hpp
index 63353a1..d2375b1 100644
--- a/test/controller_mock.hpp
+++ b/test/controller_mock.hpp
@@ -1,6 +1,7 @@
 #pragma once
 
 #include "pid/controller.hpp"
+#include "pid/zone_interface.hpp"
 
 #include <gmock/gmock.h>
 
@@ -12,7 +13,7 @@
   public:
     virtual ~ControllerMock() = default;
 
-    ControllerMock(const std::string& id, DbusPidZone* owner) :
+    ControllerMock(const std::string& id, ZoneInterface* owner) :
         PIDController(id, owner)
     {}