1 files changed, 2 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/DHT/Query.hs b/src/Network/BitTorrent/DHT/Query.hs
index 90cdcbee..1c4df267 100644
--- a/src/Network/BitTorrent/DHT/Query.hs
+++ b/src/Network/BitTorrent/DHT/Query.hs
@@ -82,7 +82,7 @@ import Data.Time
 import Data.Time.Clock.POSIX
 
 import Network.DatagramServer as KRPC hiding (Options, def)
-import Network.KRPC.Message (ReflectedIP(..))
+import Network.DatagramServer.Mainline (ReflectedIP(..))
 import Network.DatagramServer (QueryFailure(..))
 import Data.Torrent
 import Network.BitTorrent.DHT.Message
@@ -93,7 +93,7 @@ import qualified Network.BitTorrent.DHT.Search as Search
 #ifdef VERSION_bencoding
 import Data.BEncode (BValue)
 import Network.DHT.Mainline hiding (NodeId)
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 #else
 import Data.ByteString (ByteString)
 import Data.Tox