1 files changed, 11 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index 38b3ed11..bc5c6201 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -10,6 +10,7 @@
 --   to. Normally, you don't need to import this module, use
 --   "Network.BitTorrent.DHT" instead.
 --
+{-# LANGUAGE CPP                        #-}
 {-# LANGUAGE RecordWildCards            #-}
 {-# LANGUAGE FlexibleContexts           #-}
 {-# LANGUAGE FlexibleInstances          #-}
@@ -258,10 +259,20 @@ newtype DHT ip a = DHT { unDHT :: ReaderT (Node ip) IO a }
            , MonadBase IO, MonadReader (Node ip), MonadThrow
            )
 
+#if MIN_VERSION_monad_control(1,0,0)
+newtype DHTStM ip a = StM {
+      unSt :: StM (ReaderT (Node ip) IO) a
+    }
+#endif
+
 instance MonadBaseControl IO (DHT ip) where
+#if MIN_VERSION_monad_control(1,0,0)
+  type StM (DHT ip) a = DHTStM ip a
+#else
   newtype StM (DHT ip) a = StM {
       unSt :: StM (ReaderT (Node ip) IO) a
     }
+#endif
   liftBaseWith cc = DHT $ liftBaseWith $ \ cc' ->
       cc $ \ (DHT m) -> StM <$> cc' m
   {-# INLINE liftBaseWith #-}