1 files changed, 9 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index 51d92127..d9328cea 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -70,8 +70,11 @@ import Network.BitTorrent.DHT.Query
 import Network.BitTorrent.DHT.Session
 import Network.DHT.Routing as T hiding (null)
 import qualified Data.Text as Text
+import Data.Typeable
 import Data.Monoid
 import Network.DatagramServer.Mainline (KMessageOf)
+import qualified Network.DatagramServer as KRPC (listen, Protocol(..))
+
 
 
 {-----------------------------------------------------------------------
@@ -97,8 +100,12 @@ dht :: (Ord ip, Address ip)
     -> IO a        -- ^ result.
 dht opts addr logfilter action = do
   runStderrLoggingT $ filterLogger logfilter $ LoggingT $ \ logger -> do
-    bracket (newNode defaultHandlers opts addr logger Nothing) closeNode $
-      \ node -> runDHT node action
+    bracket (newNode opts addr logger Nothing) closeNode $
+      \ node -> runDHT node $ do
+            hs <- defaultHandlers logger
+            m <- asks manager
+            liftIO $ KRPC.listen m hs (KRPC.Protocol Proxy Proxy)
+            action
 {-# INLINE dht #-}
 
 {-----------------------------------------------------------------------