--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -3,7 +3,7 @@ import shutil
 import tempfile
 import unittest
 
-from mock import patch, Mock
+from unittest.mock import patch, Mock
 from six.moves.urllib.parse import quote
 
 from pgxnclient.tar import TarArchive
