Safer exception handling in probeNode.

1 files changed, 2 insertions, 1 deletions

diff --git a/src/Network/BitTorrent/DHT/Query.hs b/src/Network/BitTorrent/DHT/Query.hs
index e0338572..8215c95d 100644
--- a/src/Network/BitTorrent/DHT/Query.hs
+++ b/src/Network/BitTorrent/DHT/Query.hs
@@ -75,6 +75,7 @@ import Data.Time.Clock.POSIX
 
 import Network.KRPC hiding (Options, def)
 import Network.KRPC.Message (ReflectedIP(..))
+import Network.KRPC.Manager (QueryFailure(..))
 import Data.Torrent
 import Network.BitTorrent.Address
 import Network.BitTorrent.DHT.Message
@@ -206,7 +207,7 @@ probeNode :: Address ip => NodeAddr ip -> DHT ip (Bool, Maybe ReflectedIP)
 probeNode addr = do
   $(logDebugS) "routing.questionable_node" (T.pack (render (pPrint addr)))
   result <- try $ pingQ addr
-  let _ = fmap (const ()) result :: Either SomeException ()
+  let _ = fmap (const ()) result :: Either QueryFailure ()
   return $ either (const (False,Nothing)) (\(_,mip)->(True,mip)) result