import unittest.mock as mock

1 files changed, 5 insertions, 1 deletions

diff --git a/tests/tests.py b/tests/tests.py
index 4c7f7a4..13b1e49 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -95,7 +95,11 @@ from pokemontools.crystal import (
 )
 
 import unittest
-import mock
+
+try:
+    import unittest.mock as mock
+except ImportError:
+    import mock
 
 class TestCram(unittest.TestCase):
     "this is where i cram all of my unit tests together"