1 files changed, 8 insertions, 3 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index cdc332ab..3b5d0a90 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -41,14 +41,19 @@ library
                      , Network.BitTorrent.Peer
 
                      , Network.BitTorrent.Tracker
-                     , Network.BitTorrent.Tracker.Protocol
                      , Network.BitTorrent.Tracker.Scrape
 
                      , Network.BitTorrent.Exchange
                      , Network.BitTorrent.Exchange.Selection
-                     , Network.BitTorrent.Exchange.Protocol
 
-  other-modules:
+  if flag(testing)
+     exposed-modules:  Network.BitTorrent.Exchange.Protocol
+                     , Network.BitTorrent.Tracker.Protocol
+
+  if !flag(testing)
+     other-modules:
+                       Network.BitTorrent.Exchange.Protocol
+                     , Network.BitTorrent.Tracker.Protocol
 
 
   build-depends: