diff --git a/multipart/tests/test_multipart.py b/multipart/tests/test_multipart.py
index 5769905..7913cd2 100644
--- a/multipart/tests/test_multipart.py
+++ b/multipart/tests/test_multipart.py
@@ -16,7 +16,10 @@
 from io import BytesIO
 from six import binary_type, text_type
 
-from mock import MagicMock, Mock, patch
+try:
+    from unittest.mock import MagicMock, Mock, patch
+except ImportError:
+    from mock import MagicMock, Mock, patch
 
 from ..multipart import *