1 files changed, 2 insertions, 0 deletions
diff --git a/tests/__init__.py b/tests/__init__.py
index 9bb5ef6..cd721ae 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -29,9 +29,11 @@ testing_config.read(['test.cfg'])
 
 if py33:
     from unittest.mock import Mock, call
+    from unittest import mock
 else:
     try:
         from mock import Mock, call
+        import mock
     except ImportError:
         raise ImportError(
                 "Alembic's test suite requires the "