diff --git a/tests/datastore/testmemoryds.nim b/tests/datastore/testmemoryds.nim
index fedb497..2100f69 100644
--- a/tests/datastore/testmemoryds.nim
+++ b/tests/datastore/testmemoryds.nim
@@ -3,7 +3,7 @@ import std/sequtils
 import std/os
 from std/algorithm import sort, reversed
 
-import pkg/asynctest/unittest2
+import pkg/asynctest
 import pkg/chronos
 import pkg/stew/results
 import pkg/stew/byteutils
diff --git a/tests/datastore/testthreadproxyds.nim b/tests/datastore/testthreadproxyds.nim
index 30b91fc..0641570 100644
--- a/tests/datastore/testthreadproxyds.nim
+++ b/tests/datastore/testthreadproxyds.nim
@@ -3,7 +3,7 @@ import std/sequtils
 import std/os
 import std/algorithm
 
-import pkg/asynctest/unittest2
+import pkg/asynctest
 import pkg/chronos
 import pkg/stew/results
 import pkg/stew/byteutils