3 files changed, 5 insertions, 6 deletions

diff --git a/src/mailman_pgp/model/tests/test_db_key.py b/src/mailman_pgp/model/tests/test_db_key.py
index 46cc487..3041f0c 100644
--- a/src/mailman_pgp/model/tests/test_db_key.py
+++ b/src/mailman_pgp/model/tests/test_db_key.py
@@ -16,7 +16,7 @@
 # this program.  If not, see <http://www.gnu.org/licenses/>.
 
 """"""
-from unittest import TestCase
+import unittest
 
 from mailman_pgp.config import config
 from mailman_pgp.database import transaction
@@ -25,7 +25,7 @@ from mailman_pgp.testing.layers import PGPConfigLayer
 from mailman_pgp.testing.pgp import load_key
 
 
-class TestDBKey(TestCase):
+class TestDBKey(unittest.TestCase):
     layer = PGPConfigLayer
 
     def test_key(self):
diff --git a/src/mailman_pgp/model/tests/test_fs_key.py b/src/mailman_pgp/model/tests/test_fs_key.py
index 18c9a29..61af097 100644
--- a/src/mailman_pgp/model/tests/test_fs_key.py
+++ b/src/mailman_pgp/model/tests/test_fs_key.py
@@ -17,16 +17,16 @@
 
 """"""
 import time
+import unittest
 from os.path import exists, join
 from tempfile import TemporaryDirectory
-from unittest import TestCase
 
 from mailman_pgp.model.fs_key import FSKey
 from mailman_pgp.testing.layers import PGPLayer
 from mailman_pgp.testing.pgp import load_key
 
 
-class TestFSKey(TestCase):
+class TestFSKey(unittest.TestCase):
     layer = PGPLayer
 
     def setUp(self):
diff --git a/src/mailman_pgp/model/tests/test_list.py b/src/mailman_pgp/model/tests/test_list.py
index eb43355..e437607 100644
--- a/src/mailman_pgp/model/tests/test_list.py
+++ b/src/mailman_pgp/model/tests/test_list.py
@@ -19,7 +19,6 @@
 import shutil
 import unittest
 from os.path import exists
-from unittest import TestCase
 
 from mailman.app.lifecycle import create_list
 from mailman.interfaces.listmanager import IListManager
@@ -32,7 +31,7 @@ from mailman_pgp.testing.layers import PGPConfigLayer
 from mailman_pgp.testing.pgp import load_key
 
 
-class TestPGPMailingList(TestCase):
+class TestPGPMailingList(unittest.TestCase):
     layer = PGPConfigLayer
 
     def setUp(self):