diff --git a/torba/tests/client_tests/unit/test_stream_controller.py b/torba/tests/client_tests/unit/test_stream_controller.py
index 2d33f698a..496975536 100644
--- a/torba/tests/client_tests/unit/test_stream_controller.py
+++ b/torba/tests/client_tests/unit/test_stream_controller.py
@@ -1,6 +1,3 @@
-import asyncio
-import unittest
-
 from torba.stream import StreamController
 from torba.tasks import TaskGroup
 from torba.testcase import AsyncioTestCase
@@ -24,8 +21,9 @@ class StreamControllerTestCase(AsyncioTestCase):
         self.assertListEqual(events, ["yo"])
 
 
-class TaskGroupTestCase(unittest.TestCase):
-    def test_cancel_sets_it_done(self):
+class TaskGroupTestCase(AsyncioTestCase):
+
+    async def test_cancel_sets_it_done(self):
         group = TaskGroup()
         group.cancel()
         self.assertTrue(group.done.is_set())