diff --git a/python/ql/test/query-tests/Statements/asserts/assert.py b/python/ql/test/query-tests/Statements/asserts/assert.py
index 036f9b20a4e..e4ca2bfe2bd 100644
--- a/python/ql/test/query-tests/Statements/asserts/assert.py
+++ b/python/ql/test/query-tests/Statements/asserts/assert.py
@@ -61,8 +61,8 @@ def ok_assert_false(x):
     if x:
         assert 0==1, "Ok"
 
-class TestCase:
-    pass
+from unittest import TestCase
+
 
 class MyTest(TestCase):
     def test_ok_assert_in_test(self, x):