diff --git a/tests/scripts/check_test_cases.py b/tests/scripts/check_test_cases.py
index 6809dd5e77..a7ad9aebe3 100755
--- a/tests/scripts/check_test_cases.py
+++ b/tests/scripts/check_test_cases.py
@@ -18,6 +18,7 @@ import sys
 
 import scripts_path # pylint: disable=unused-import
 from mbedtls_framework import build_tree
+import collect_test_cases
 
 class ScriptOutputError(ValueError):
     """A kind of ValueError that indicates we found
diff --git a/tests/scripts/collect_test_cases.py b/tests/scripts/collect_test_cases.py
new file mode 100644
index 0000000000..7e9031c970
--- /dev/null
+++ b/tests/scripts/collect_test_cases.py
@@ -0,0 +1,4 @@
+"""Discover all the test cases (unit tests and SSL tests)."""
+
+# Copyright The Mbed TLS Contributors
+# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later