diff --git a/warden_server/test_warden_server.py b/warden_server/test_warden_server.py
index 4d1a84c4ecbb1bb16b5233009a04c6cdacc5a01d..d690a8896f401f467f8d63f7f65ea3f69b1639e4 100755
--- a/warden_server/test_warden_server.py
+++ b/warden_server/test_warden_server.py
@@ -15,7 +15,7 @@ import warden_server
 if sys.version_info >= (3, 10):
     import unittest
 else:
-    import unittest2
+    import unittest2 as unittest
 
 if sys.version_info[0] >= 3:
     from io import StringIO