--- a/tests/unit/distributed/test_transaction.py
+++ b/tests/unit/distributed/test_transaction.py
@@ -1,5 +1,5 @@
 import pytest
-import mock
+from unittest import mock
 
 from bcbio.distributed import transaction
 from bcbio.distributed.transaction import tx_tmpdir
--- a/tests/unit/upload/test_upload.py
+++ b/tests/unit/upload/test_upload.py
@@ -1,5 +1,5 @@
 import os
-import mock
+from unittest import mock
 import pytest
 
 from bcbio import upload