1 files changed, 4 insertions, 4 deletions

diff --git a/Kademlia.hs b/Kademlia.hs
index ef5c6a48..017209a7 100644
--- a/Kademlia.hs
+++ b/Kademlia.hs
@@ -267,10 +267,11 @@ forkPollForRefresh interval psq refresh = do
 
 refreshBucket :: forall nid tok ni addr.
                  ( Show nid, FiniteBits nid, Serialize nid, Ord nid, Ord ni, Hashable nid, Hashable ni, Ord addr ) =>
-                 Search nid addr tok ni ni -> TVar (BucketList ni) -> nid -> Int -> IO Int
-refreshBucket sch var nid n = do
+                 Search nid addr tok ni ni -> TVar (BucketList ni) -> Int -> IO Int
+refreshBucket sch var n = do
     tbl <- atomically (readTVar var)
     let count = bktCount tbl
+        nid = kademliaLocation (searchSpace sch) (thisNode tbl)
     sample <- if n+1 >= count            -- Is this the last bucket?
                 then return nid          -- Yes? Search our own id.
                 else genBucketSample nid -- No? Generate a random id.
@@ -362,7 +363,6 @@ bootstrap sch var ping ns ns0 = do
                     hPutStrLn stderr
                         $ "Not enough buckets, refresh " ++ show (R.defaultBucketCount - 1)
                     cnt <- refreshBucket sch var
-                        (kademliaLocation (searchSpace sch) (thisNode tbl))
                         (R.defaultBucketCount - 1)
                     again cnt
             (size,num):_ -> do
@@ -371,5 +371,5 @@ bootstrap sch var ping ns ns0 = do
                 let num' | bktCount tbl < R.defaultBucketCount = R.defaultBucketCount - 1
                          | otherwise                           = num
                 hPutStrLn stderr $ "Bucket too small, refresh "++ show (num',(size,num),shp)
-                cnt <- refreshBucket sch var (kademliaLocation (searchSpace sch) (thisNode tbl)) num'
+                cnt <- refreshBucket sch var num'
                 again cnt