diff --git a/tests/testidentify.nim b/tests/testidentify.nim
index 043c728bd..1f561d86e 100644
--- a/tests/testidentify.nim
+++ b/tests/testidentify.nim
@@ -1,10 +1,14 @@
 import unittest, options
 import chronos, strutils, sequtils
-import ../libp2p/identify, ../libp2p/multiaddress, 
-       ../libp2p/peerinfo, ../libp2p/peer, 
-       ../libp2p/connection, ../libp2p/identify, 
-       ../libp2p/multistream, ../libp2p/transport,
-       ../libp2p/tcptransport, ../libp2p/protocol, 
+import ../libp2p/protocols/identify, 
+       ../libp2p/multiaddress, 
+       ../libp2p/peerinfo, 
+       ../libp2p/peer, 
+       ../libp2p/connection, 
+       ../libp2p/multistream, 
+       ../libp2p/transports/transport,
+       ../libp2p/transports/tcptransport, 
+       ../libp2p/protocols/protocol, 
        ../libp2p/crypto/crypto
 
 suite "Identify":
diff --git a/tests/testtransport.nim b/tests/testtransport.nim
index 0091de1a5..4d4bf3b51 100644
--- a/tests/testtransport.nim
+++ b/tests/testtransport.nim
@@ -1,7 +1,10 @@
 import unittest
 import chronos
-import ../libp2p/connection, ../libp2p/transport, ../libp2p/tcptransport,
-    ../libp2p/multiaddress, ../libp2p/wire
+import ../libp2p/connection, 
+       ../libp2p/transports/transport, 
+       ../libp2p/transports/tcptransport,
+       ../libp2p/multiaddress, 
+       ../libp2p/wire
 
 suite "TCP transport":
   test "test listener: handle write":