1 files changed, 7 insertions, 6 deletions

diff --git a/dht/examples/testTox.hs b/dht/examples/testTox.hs
index 6db977be..36162163 100644
--- a/dht/examples/testTox.hs
+++ b/dht/examples/testTox.hs
@@ -13,8 +13,11 @@ import Control.Concurrent.STM.TVar
 import Control.Monad
 import Control.Monad.STM
 import Crypto.Tox
+import Data.Dependent.Sum
 import qualified Data.IntMap.Strict          as IntMap
 import Data.Function
+import Data.Tox.Msg
+import qualified Data.Tox.DHT.Multi          as Multi
 import DebugUtil
 import DPut
 import DebugTag
@@ -34,8 +37,6 @@ import qualified Data.HashMap.Strict as HashMap
 import qualified Data.Map.Strict as Map
 import Data.Time.Clock.POSIX
 import System.Exit
-import Data.Dependent.Sum
-import Data.Tox.Msg
 
 makeToxNode :: UDPTransport -> Maybe SecretKey
         -> ( ContactInfo extra -> SockAddr -> Session -> IO () )
@@ -74,11 +75,11 @@ sessionChan remotes tchan acnt saddr s = do
 
 netCrypto :: Tox extra -> SecretKey -> NodeInfo -> PublicKey -> IO ()
 netCrypto tox me ni them = do
-    mcookie <- cookieRequest (toxCryptoKeys tox) (toxDHT tox) (toPublic me) ni
+    mcookie <- cookieRequest (toxCryptoKeys tox) (toxDHT tox) (toPublic me) (Multi.UDP ==> ni)
     case mcookie of
         Just cookie -> do
-            hs <- cacheHandshake (toxHandshakeCache tox) me them ni cookie
-            sendMessage (toxHandshakes tox) (nodeAddr ni) hs
+            hs <- cacheHandshake (toxHandshakeCache tox) me them (Multi.UDP ==> ni) cookie
+            sendMessage (toxHandshakes tox) (Multi.SessionUDP ==> nodeAddr ni) hs
         Nothing -> do
             dput XUnused "Timeout requesting cookie."
 
@@ -145,7 +146,7 @@ main = do
                         }
                       )
 
-    DHT.ping (toxDHT a_Ct7g5azVcJ8KnvxzbXs9GqeqcjrP7VdZXIC'uHeTlRf) b
+    DHT.pingUDP (toxDHT a_Ct7g5azVcJ8KnvxzbXs9GqeqcjrP7VdZXIC'uHeTlRf) b
 
     -- sendMessage (toxHandshakes a_Ct7g5azVcJ8KnvxzbXs9GqeqcjrP7VdZXIC'uHeTlRf) (nodeAddr b) hs