7 files changed, 8 insertions, 8 deletions

diff --git a/src/Network/BitTorrent/Address.hs b/src/Network/BitTorrent/Address.hs
index 05c6504c..e75066b1 100644
--- a/src/Network/BitTorrent/Address.hs
+++ b/src/Network/BitTorrent/Address.hs
@@ -128,7 +128,7 @@ import System.Locale    (defaultTimeLocale)
 import System.Entropy
 import Data.Digest.CRC32C
 import Network.DatagramServer.Types as RPC
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 -- import Network.DHT.Mainline
 
 -- import Paths_bittorrent (version)
diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index ab948a2d..169abcfc 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -71,7 +71,7 @@ import Network.BitTorrent.DHT.Session
 import Network.BitTorrent.DHT.Routing as T hiding (null)
 import qualified Data.Text as Text
 import Data.Monoid
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 
 
 {-----------------------------------------------------------------------
diff --git a/src/Network/BitTorrent/DHT/Message.hs b/src/Network/BitTorrent/DHT/Message.hs
index 2a2cee35..cfe24fc5 100644
--- a/src/Network/BitTorrent/DHT/Message.hs
+++ b/src/Network/BitTorrent/DHT/Message.hs
@@ -108,7 +108,7 @@ import Data.Serialize as S
 import Data.Typeable
 import Network
 import Network.DatagramServer
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 import Data.Maybe
 
 import Data.Torrent (InfoHash)
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
diff --git a/src/Network/BitTorrent/DHT/Routing.hs b/src/Network/BitTorrent/DHT/Routing.hs
index c64678e0..d147bb94 100644
--- a/src/Network/BitTorrent/DHT/Routing.hs
+++ b/src/Network/BitTorrent/DHT/Routing.hs
@@ -87,7 +87,7 @@ import Data.Torrent
 import Network.BitTorrent.Address
 #ifdef VERSION_bencoding
 import Network.DHT.Mainline ()
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 #else
 import Data.Tox as Tox
 type KMessageOf = Tox.Message
diff --git a/src/Network/BitTorrent/DHT/Search.hs b/src/Network/BitTorrent/DHT/Search.hs
index c60ae48f..7a587644 100644
--- a/src/Network/BitTorrent/DHT/Search.hs
+++ b/src/Network/BitTorrent/DHT/Search.hs
@@ -27,7 +27,7 @@ import Network.BitTorrent.Address hiding (NodeId)
 import Network.DatagramServer.Types
 #ifdef VERSION_bencoding
 import Network.DHT.Mainline ()
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 type Ann = ()
 #else
 import Data.Tox as Tox
diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index 0800f147..ead4a723 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -108,7 +108,7 @@ import Network.DatagramServer as KRPC hiding (Options, def)
 import qualified Network.DatagramServer as KRPC (def)
 #ifdef VERSION_bencoding
 import Data.BEncode (BValue)
-import Network.KRPC.Message (KMessageOf)
+import Network.DatagramServer.Mainline (KMessageOf)
 #else
 import Data.Tox as Tox
 #endif