1 files changed, 10 insertions, 1 deletions
diff --git a/src/usb-manager.h b/src/usb-manager.h
index ec405c0..960d634 100644
--- a/src/usb-manager.h
+++ b/src/usb-manager.h
@@ -19,6 +19,8 @@
 
 #pragma once
 
+#include <src/usb-monitor.h>
+
 #include <memory>
 #include <string>
 
@@ -28,10 +30,17 @@
 class UsbManager
 {
 public:
-    UsbManager(const std::string& socket_path, const std::string& public_key_filename);
+
+    UsbManager(
+        const std::string& socket_path,
+        const std::string& public_key_filename,
+        const std::shared_ptr<UsbMonitor>&
+    );
+
     ~UsbManager();
 
 protected:
+
     class Impl;
     std::unique_ptr<Impl> impl;
 };