diff --git a/tests/test_web_user.py b/tests/test_web_user.py
index 5585fb5e4d..9be013dfe6 100644
--- a/tests/test_web_user.py
+++ b/tests/test_web_user.py
@@ -3,7 +3,10 @@
 
 import unittest
 
-from mock import patch, ANY
+try:
+    from unittest.mock import patch, ANY
+except ImportError:
+    from mock import patch, ANY
 
 from trytond.tests.test_tryton import ModuleTestCase, with_transaction
 from trytond.tests.test_tryton import suite as test_suite