diff --git a/tools/common.h b/tools/common.h
index 97c44c9..dbd72c3 100644
--- a/tools/common.h
+++ b/tools/common.h
@@ -21,6 +21,9 @@
 #include "ice_transport_factory.h"
 #include "certstore.h"
 
+#include "upnp/upnp_control.h"
+#include "upnp/upnp_context.h"
+
 namespace dhtnet {
 
 #define Log(...) do { fmt::print(__VA_ARGS__); std::fflush(stdout); } while (0)
@@ -40,7 +43,8 @@
     const std::string& turn_host ="",
     const std::string& turn_user="",
     const std::string& turn_pass="",
-    const std::string& turn_realm="");
+    const std::string& turn_realm="",
+    const bool enable_upnp=true);
 // add ioContext to readFromStdin
 
 template<typename T>