diff --git a/src/wg/relayreg.h b/src/wg/relayreg.h
index 3e214f6..583a487 100644
--- a/src/wg/relayreg.h
+++ b/src/wg/relayreg.h
@@ -11,8 +11,11 @@
 // Built on libcurl (synchronous, no event loop dependency).
 // The caller is responsible for starting heartbeats in a background thread.
 
+#include "displayinfo.h"
+
 #include <functional>
 #include <string>
+#include <vector>
 
 namespace wg {
 
@@ -63,8 +66,10 @@ public:
     // name: friendly display name shown in DragonMoonlight.
     // wgIP: the local WireGuard address (e.g. "10.99.0.3").
     // port: streaming port (default 47984).
+    // displays: list of connected display info.
     bool registerHost(const std::string &name, const std::string &wgIP,
-                      int port, std::string &errOut);
+                      int port, const std::vector<DisplayInfo> &displays,
+                      std::string &errOut);
 
     // Must be called every ~60 seconds to keep the host visible (5-min TTL).
     bool heartbeat(std::string &errOut);