diff --git a/tests/rest/test_args.py b/tests/rest/test_args.py
index 4ae3246..060ac0b 100644
--- a/tests/rest/test_args.py
+++ b/tests/rest/test_args.py
@@ -1,4 +1,4 @@
-import mock
+from unittest import mock
 import unittest
 
 from wsme import exc