--- a/py_vapid/tests/test_vapid.py
+++ b/py_vapid/tests/test_vapid.py
@@ -5,7 +5,7 @@
 import json
 import unittest
 from cryptography.hazmat.primitives import serialization
-from mock import patch, Mock
+from unittest.mock import patch, Mock
 
 from py_vapid import Vapid01, Vapid02, VapidException, _check_sub
 from py_vapid.jwt import decode