diff --git a/tests/test_utility_functions.py b/tests/test_utility_functions.py
index c9fffd3..c01b0bc 100644
--- a/tests/test_utility_functions.py
+++ b/tests/test_utility_functions.py
@@ -36,9 +36,13 @@
     - is_some_sort_of_code()
 """
 
+try:
+    from unittest.mock import patch
+except ImportError
+    from mock import patch
+
 # Third Party Imports
 import pytest
-from mock import patch
 
 # docformatter Package Imports
 import docformatter