1 files changed, 3 insertions, 1 deletions

diff --git a/tests/Network/BitTorrent/DHT/RoutingSpec.hs b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
index a92fec38..651412cf 100644
--- a/tests/Network/BitTorrent/DHT/RoutingSpec.hs
+++ b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
@@ -31,7 +31,9 @@ runSimulation e m = evalState (runRouting ping closest timestamp m) e
     timestamp    = gets currentTime
 
 instance Arbitrary ip => Arbitrary (Env ip) where
-  arbitrary = Env <$> arbitrary <*> arbitrary
+  arbitrary = Env <$> arbitrary <*> (L.take nodeLimit <$> arbitrary)
+    where
+      nodeLimit = 1000
 
 instance (Arbitrary ip, Eq ip) => Arbitrary (Table ip) where
   arbitrary = do