1 files changed, 3 insertions, 5 deletions

diff --git a/src/Network/Tox/DHT/Transport.hs b/src/Network/Tox/DHT/Transport.hs
index e784ec90..b9b63165 100644
--- a/src/Network/Tox/DHT/Transport.hs
+++ b/src/Network/Tox/DHT/Transport.hs
@@ -382,13 +382,11 @@ instance Sized CookieData where
     size = ConstSize 72
 
 instance Serialize CookieData where
-    get = CookieData <$> get
-                     <*> (id2key <$> get)
-                     <*> (id2key <$> get)
+    get = CookieData <$> get <*> getPublicKey <*> getPublicKey
     put (CookieData tm userkey dhtkey) = do
         put tm
-        put (key2id userkey)
-        put (key2id dhtkey)
+        putPublicKey userkey
+        putPublicKey userkey
 
 instance Sized CookieRequest where
     size = ConstSize 64 -- 32 byte key + 32 byte padding