diff --git a/tests/test_octodns_source_envvar.py b/tests/test_octodns_source_envvar.py
index 38b63dd..ac66a22 100644
--- a/tests/test_octodns_source_envvar.py
+++ b/tests/test_octodns_source_envvar.py
@@ -1,13 +1,7 @@
+from mock import patch
 from six import text_type
 from unittest import TestCase
 
-try:
-    # Python 3
-    from unittest.mock import patch
-except ImportError:
-    # Python 2
-    from mock import patch
-
 from octodns.source.envvar import EnvVarSource
 from octodns.source.envvar import EnvironmentVariableNotFoundException
 from octodns.zone import Zone