diff --git a/tests/test_builder/test_newRepo.py b/tests/test_builder/test_newRepo.py
index d832a0ab..cf6a9dd1 100644
--- a/tests/test_builder/test_newRepo.py
+++ b/tests/test_builder/test_newRepo.py
@@ -1,5 +1,8 @@
 from __future__ import absolute_import
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import os
 import rpm
 import shutil
diff --git a/tests/test_kojira/test_main.py b/tests/test_kojira/test_main.py
index beb217d3..2531d274 100644
--- a/tests/test_kojira/test_main.py
+++ b/tests/test_kojira/test_main.py
@@ -7,7 +7,10 @@ import tempfile
 import time
 import unittest
 
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import pytest
 
 import koji
diff --git a/tests/test_plugins/test_builder_scmpolicy.py b/tests/test_plugins/test_builder_scmpolicy.py
index b99645ed..e6e9fc29 100644
--- a/tests/test_plugins/test_builder_scmpolicy.py
+++ b/tests/test_plugins/test_builder_scmpolicy.py
@@ -1,5 +1,8 @@
 import unittest
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 
 import scmpolicy
 import koji