diff --git a/tests/datastore/testdatastore.nim b/tests/datastore/testdatastore.nim
index 4361eef..cd52915 100644
--- a/tests/datastore/testdatastore.nim
+++ b/tests/datastore/testdatastore.nim
@@ -25,5 +25,6 @@ suite "Datastore (base)":
 
   test "query":
     expect Defect:
-      let iter = (await ds.query(Query.init(key))).tryGet
-      for n in iter: discard
+      let iter = tryGet(await waitForAllQueryResults ds.query(Query.init(key)))
+      for n in iter:
+        discard