d20a0bc
diff --git a/tests/test_pid.py b/tests/test_pid.py
d20a0bc
index 7c57c46..e36c29e 100644
d20a0bc
--- a/tests/test_pid.py
d20a0bc
+++ b/tests/test_pid.py
d20a0bc
@@ -2,7 +2,10 @@
d20a0bc
 import os.path
d20a0bc
 import signal
d20a0bc
 from contextlib import contextmanager
d20a0bc
-from mock import patch
d20a0bc
+try:
d20a0bc
+    from unittest.mock import patch
d20a0bc
+except ImportError:
d20a0bc
+    from mock import patch
d20a0bc
 
d20a0bc
 import pid
d20a0bc