diff --git a/warden_server/test_warden_server.py b/warden_server/test_warden_server.py
index c569230ae9e1a4953b37d6fbdfb7613a579af801..bee3433cf9627751788fb6c779914846b5133398 100755
--- a/warden_server/test_warden_server.py
+++ b/warden_server/test_warden_server.py
@@ -8,11 +8,15 @@ import sys
 import warnings
 from os import path
 from copy import deepcopy
-import unittest2 as unittest
 import MySQLdb as my
 from warden_server import build_server
 import warden_server
 
+if sys.version_info >= (3, 10):
+    import unittest
+else:
+    import unittest2
+
 if sys.version_info[0] >= 3:
     from io import StringIO
 else: