1 files changed, 4 insertions, 3 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index b63bd6fc..9bb7c5a7 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -53,7 +53,7 @@ import Network.DHT.Routing as R
 import Data.Aeson as J (ToJSON, FromJSON)
 import qualified Data.Aeson as J
 import qualified Data.ByteString.Lazy as L
-import Control.Concurrent.Async.Pool
+import Tasks
 import System.IO.Error
 import qualified Data.Serialize as S
 import Network.BitTorrent.DHT.ContactInfo as Peers
@@ -139,8 +139,9 @@ pingNodes netname DHT{dhtPing} = do
         Right ns -> do fork $ do
                          myThreadId >>= flip labelThread ("pinging."++fname)
                          putStrLn $ "Forked "++show fname
-                         withTaskGroup 10 $ \g -> do
-                           mapTasks_ g (map dhtPing ns)
+                         withTaskGroup ("withTaskGroup."++fname) 10 $ \g -> do
+                           mapM_ (\n -> forkTask g (show n) $ void $ dhtPing n)
+                                 (ns `asTypeOf` [])
                          putStrLn $ "Load finished "++show fname
                        return True