--- a/test/test_genty.py
+++ b/test/test_genty.py
@@ -3,7 +3,10 @@
 from __future__ import unicode_literals
 import functools
 import inspect
-from mock import patch
+try:
+    from unittest.mock import patch
+except ImportError:
+    from mock import patch
 import six
 from genty import genty, genty_args, genty_dataset, genty_repeat, genty_dataprovider
 from genty.genty import REPLACE_FOR_PERIOD_CHAR