diff --git a/tests/pubsub/testfloodsub.nim b/tests/pubsub/testfloodsub.nim
index e7e2f31..d60b30b 100644
--- a/tests/pubsub/testfloodsub.nim
+++ b/tests/pubsub/testfloodsub.nim
@@ -7,7 +7,7 @@
 ## This file may not be copied, modified, or distributed except according to
 ## those terms.
 
-import unittest, sequtils, options
+import unittest, sequtils
 import chronos
 import utils,
        ../../libp2p/[switch,
diff --git a/tests/testswitch.nim b/tests/testswitch.nim
index fcbf248..ce689ef 100644
--- a/tests/testswitch.nim
+++ b/tests/testswitch.nim
@@ -101,6 +101,7 @@ suite "Switch":
 
       check isNil(conn)
       discard allFutures(switch1.stop(), switch2.stop())
+      await allFutures(awaiters)
       result = true
 
     check: