diff --git a/Redirector/DNSHandler.cpp b/Redirector/DNSHandler.cpp
index bb214f29..48e390f8 100644
--- a/Redirector/DNSHandler.cpp
+++ b/Redirector/DNSHandler.cpp
@@ -1,5 +1,7 @@
 #include "DNSHandler.h"
 
+#include "SocksHelper.h"
+
 extern string dnsHost;
 extern USHORT dnsPort;
 
diff --git a/Redirector/DNSHandler.h b/Redirector/DNSHandler.h
index e7548052..9800cbea 100644
--- a/Redirector/DNSHandler.h
+++ b/Redirector/DNSHandler.h
@@ -2,7 +2,6 @@
 #ifndef DNSHANDLER_H
 #define DNSHANDLER_H
 #include "Based.h"
-#include "SocksHelper.h"
 
 namespace DNSHandler
 {
diff --git a/Redirector/EventHandler.h b/Redirector/EventHandler.h
index f84c70a2..c07e2c2c 100644
--- a/Redirector/EventHandler.h
+++ b/Redirector/EventHandler.h
@@ -24,6 +24,6 @@ void udpCanReceive(ENDPOINT_ID id);
 void udpReceive(ENDPOINT_ID id, const unsigned char* target, const char* buffer, int length, PNF_UDP_OPTIONS options);
 void udpClosed(ENDPOINT_ID id, PNF_UDP_CONN_INFO info);
 
-void udpReceiveHandler(ENDPOINT_ID id, SocksHelper::PUDP conn, PNF_UDP_OPTIONS options);
+void udpReceiveHandler(ENDPOINT_ID id, SocksHelper::PUDP remote, PNF_UDP_OPTIONS options);
 
 #endif