diff --git a/tests/integration/test_validators.py b/tests/integration/test_validators.py
index 6ab8f8c..246133d 100644
--- a/tests/integration/test_validators.py
+++ b/tests/integration/test_validators.py
@@ -1,10 +1,14 @@
 from jsonschema import ValidationError
-import mock
 import pytest
 from six import u
 
 from openapi_schema_validator import OAS30Validator, oas30_format_checker
 
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
 
 class TestOAS30ValidatorValidate(object):