diff -Naur pgcli-3.1.0-original/tests/metadata.py pgcli-3.1.0/tests/metadata.py
--- pgcli-3.1.0-original/tests/metadata.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/metadata.py	2021-03-29 12:48:22.105962073 -0400
@@ -3,7 +3,7 @@
 from pgcli.packages.parseutils.meta import FunctionMetadata, ForeignKey
 from prompt_toolkit.completion import Completion
 from prompt_toolkit.document import Document
-from mock import Mock
+from unittest.mock import Mock
 import pytest
 
 parametrize = pytest.mark.parametrize
diff -Naur pgcli-3.1.0-original/tests/test_completion_refresher.py pgcli-3.1.0/tests/test_completion_refresher.py
--- pgcli-3.1.0-original/tests/test_completion_refresher.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_completion_refresher.py	2021-03-29 12:48:25.523982262 -0400
@@ -1,6 +1,6 @@
 import time
 import pytest
-from mock import Mock, patch
+from unittest.mock import Mock, patch
 
 
 @pytest.fixture
diff -Naur pgcli-3.1.0-original/tests/test_main.py pgcli-3.1.0/tests/test_main.py
--- pgcli-3.1.0-original/tests/test_main.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_main.py	2021-03-29 12:48:01.263838960 -0400
@@ -1,6 +1,6 @@
 import os
 import platform
-import mock
+from unittest import mock
 
 import pytest
 
diff -Naur pgcli-3.1.0-original/tests/test_naive_completion.py pgcli-3.1.0/tests/test_naive_completion.py
--- pgcli-3.1.0-original/tests/test_naive_completion.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_naive_completion.py	2021-03-29 12:48:28.710001082 -0400
@@ -13,7 +13,7 @@
 
 @pytest.fixture
 def complete_event():
-    from mock import Mock
+    from unittest.mock import Mock
 
     return Mock()
 
diff -Naur pgcli-3.1.0-original/tests/test_pgexecute.py pgcli-3.1.0/tests/test_pgexecute.py
--- pgcli-3.1.0-original/tests/test_pgexecute.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_pgexecute.py	2021-03-29 12:47:49.041766766 -0400
@@ -2,7 +2,7 @@
 
 import psycopg2
 import pytest
-from mock import patch, MagicMock
+from unittest.mock import patch, MagicMock
 from pgspecial.main import PGSpecial, NO_QUERY
 from utils import run, dbtest, requires_json, requires_jsonb
 
diff -Naur pgcli-3.1.0-original/tests/test_rowlimit.py pgcli-3.1.0/tests/test_rowlimit.py
--- pgcli-3.1.0-original/tests/test_rowlimit.py	2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_rowlimit.py	2021-03-29 12:48:06.210868182 -0400
@@ -1,5 +1,5 @@
 import pytest
-from mock import Mock
+from unittest.mock import Mock
 
 from pgcli.main import PGCli