--- a/tests/unit/test_handler.py
+++ b/tests/unit/test_handler.py
@@ -17,7 +17,7 @@
 import sys
 import zlib
 
-import mock
+from unittest import mock
 import pytest
 
 from graypy.handler import BaseGELFHandler, GELFHTTPHandler, GELFTLSHandler