diff --git a/castellan/tests/unit/key_manager/test_barbican_key_manager.py b/castellan/tests/unit/key_manager/test_barbican_key_manager.py
index 36e842d..4d6c9bb 100644
--- a/castellan/tests/unit/key_manager/test_barbican_key_manager.py
+++ b/castellan/tests/unit/key_manager/test_barbican_key_manager.py
@@ -17,9 +17,9 @@
 Test cases for the barbican key manager.
 """
 import calendar
+from unittest import mock
 
 from barbicanclient import exceptions as barbican_exceptions
-import mock
 from oslo_utils import timeutils
 
 from castellan.common import exception
diff --git a/castellan/tests/unit/key_manager/test_migration_key_manager.py b/castellan/tests/unit/key_manager/test_migration_key_manager.py
index e5c8ba6..ffba620 100644
--- a/castellan/tests/unit/key_manager/test_migration_key_manager.py
+++ b/castellan/tests/unit/key_manager/test_migration_key_manager.py
@@ -18,7 +18,7 @@
 """
 
 import binascii
-import mock
+from unittest import mock
 
 from oslo_config import cfg