diff --git a/tests/basic/test_commands.py b/tests/basic/test_commands.py
index 61be09522..003afc3c6 100644
--- a/tests/basic/test_commands.py
+++ b/tests/basic/test_commands.py
@@ -3,6 +3,7 @@ import os
 import shutil
 import sys
 import tempfile
+import unittest
 from io import StringIO
 from pathlib import Path
 from unittest import TestCase