diff --git a/coverage2sql/tests/test_shell.py b/coverage2sql/tests/test_shell.py
index 5e54676..f3c0ea6 100644
--- a/coverage2sql/tests/test_shell.py
+++ b/coverage2sql/tests/test_shell.py
@@ -13,8 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from unittest import mock
+
 import fixtures
-import mock
 from oslo_config import cfg
 import tempfile