1 files changed, 6 insertions, 2 deletions

diff --git a/kad/src/Network/Kademlia/CommonAPI.hs b/kad/src/Network/Kademlia/CommonAPI.hs
index 4714cecc..6d3fd16c 100644
--- a/kad/src/Network/Kademlia/CommonAPI.hs
+++ b/kad/src/Network/Kademlia/CommonAPI.hs
@@ -1,5 +1,8 @@
 {-# LANGUAGE ExistentialQuantification #-}
-module Network.Kademlia.CommonAPI where
+module Network.Kademlia.CommonAPI
+    ( module Network.Kademlia.CommonAPI
+    , refreshBuckets
+    ) where
 
 
 import Control.Concurrent
@@ -12,6 +15,7 @@ import qualified Data.Set          as Set
 import Data.Time.Clock.POSIX
 import Data.Typeable
 
+import Network.Kademlia.Bootstrap
 import Network.Kademlia.Search
 import Network.Kademlia.Routing as R
 import Crypto.Tox (SecretKey,PublicKey)
@@ -29,7 +33,7 @@ data DHT = forall nid ni. ( Show     ni
                           , S.Serialize nid
                           ) =>
     DHT
-    { dhtBuckets       :: TVar (BucketList ni)
+    { dhtBuckets       :: BucketRefresher nid ni
     , dhtSecretKey     :: STM (Maybe SecretKey)
     , dhtPing          :: Map.Map String (DHTPing ni)
     , dhtQuery         :: Map.Map String (DHTQuery nid ni)