diff --git a/tests/test_cli/test_remove_group_pkg.py b/tests/test_cli/test_remove_group_pkg.py
index df167b04..42d6398a 100644
--- a/tests/test_cli/test_remove_group_pkg.py
+++ b/tests/test_cli/test_remove_group_pkg.py
@@ -1,6 +1,9 @@
 from __future__ import absolute_import
 
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import six
 
 from koji_cli.commands import handle_remove_group_pkg
diff --git a/tests/test_cli/test_remove_group_req.py b/tests/test_cli/test_remove_group_req.py
index 79706394..8bfea48a 100644
--- a/tests/test_cli/test_remove_group_req.py
+++ b/tests/test_cli/test_remove_group_req.py
@@ -1,6 +1,9 @@
 from __future__ import absolute_import
 
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import six
 
 from koji_cli.commands import handle_remove_group_req