diff --git a/test/dbushelper_mock.hpp b/test/dbushelper_mock.hpp
index 6ce63c8..44b3bea 100644
--- a/test/dbushelper_mock.hpp
+++ b/test/dbushelper_mock.hpp
@@ -2,8 +2,6 @@
 
 #include "util.hpp"
 
-#include <sdbusplus/bus.hpp>
-
 #include <string>
 
 #include <gmock/gmock.h>
@@ -16,16 +14,13 @@
   public:
     virtual ~DbusHelperMock() = default;
 
-    MOCK_METHOD3(getService,
-                 std::string(sdbusplus::bus::bus&, const std::string&,
-                             const std::string&));
-    MOCK_METHOD4(getProperties,
-                 void(sdbusplus::bus::bus&, const std::string&,
-                      const std::string&, struct SensorProperties*));
+    MOCK_METHOD2(getService,
+                 std::string(const std::string&, const std::string&));
+    MOCK_METHOD3(getProperties, void(const std::string&, const std::string&,
+                                     struct SensorProperties*));
 
-    MOCK_METHOD3(thresholdsAsserted,
-                 bool(sdbusplus::bus::bus& bus, const std::string& service,
-                      const std::string& path));
+    MOCK_METHOD2(thresholdsAsserted,
+                 bool(const std::string& service, const std::string& path));
 };
 
 } // namespace pid_control