1 files changed, 2 insertions, 2 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index dee8bbe5..6470f612 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -77,7 +77,7 @@ library
                      , RecordWildCards
   hs-source-dirs:      src
   exposed-modules:     Network.DatagramServer
-                       Network.KRPC.Message
+                       Network.DatagramServer.Mainline
                        Network.KRPC.Method
                        Data.Torrent
                        Data.Digest.CRC32C
@@ -260,7 +260,7 @@ test-suite spec
                        Config
                        Network.KRPCSpec
                        Network.KRPC.MethodSpec
-                       Network.KRPC.MessageSpec
+                       Network.DatagramServer.MainlineSpec
                        Data.TorrentSpec
                        Network.BitTorrent.Client.HandleSpec
                        Network.BitTorrent.CoreSpec