1 files changed, 9 insertions, 6 deletions

diff --git a/tests/Network/BitTorrent/DHT/RoutingSpec.hs b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
index 3077a52a..6c0caad5 100644
--- a/tests/Network/BitTorrent/DHT/RoutingSpec.hs
+++ b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
@@ -46,14 +46,17 @@ instance (Arbitrary ip, Eq ip) => Arbitrary (Table ip) where
 --    nodes   <- vector nodeN
 
     node    <- arbitrary
-    mt <- runSimulation  <$> arbitrary
-          <*> pure (T.insert node table)
+    mt <- do
+        env <- arbitrary
+        return $ runSimulation env $ do
+            (_,t') <- T.insert (currentTime env) (TryInsert node) table
+            return t' :: Routing ip (Table ip)
           --(foldM (flip fillTable) table nodes)
     return (fromJust mt)
-   where
-    fillTable x t = do
-      t' <- T.insert x t
-      return $ if T.full t' then t else t'
+--    where
+--     fillTable x t = do
+--       t' <- T.insert x t
+--       return $ if T.full t' then t else t'
 
 spec :: Spec
 spec = do