diff --git a/oslotest/tests/unit/test_output.py b/oslotest/tests/unit/test_output.py
index 918c27f..d2605a5 100644
--- a/oslotest/tests/unit/test_output.py
+++ b/oslotest/tests/unit/test_output.py
@@ -16,7 +16,7 @@ import sys
 
 from oslotest import output
 
-import mock
+from six.moves import mock
 import testtools
 
 
diff --git a/oslotest/tests/unit/test_timeout.py b/oslotest/tests/unit/test_timeout.py
index dc16d93..1ba5bed 100644
--- a/oslotest/tests/unit/test_timeout.py
+++ b/oslotest/tests/unit/test_timeout.py
@@ -12,7 +12,7 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-import mock
+from six.moves import mock
 import testtools
 
 from oslotest import timeout