--- a/aws_requests_auth/tests/test_aws_auth.py
+++ b/aws_requests_auth/tests/test_aws_auth.py
@@ -1,6 +1,6 @@
 import datetime
 import hashlib
-import mock
+from unittest import mock
 import sys
 import unittest
 
--- a/aws_requests_auth/tests/test_boto_utils.py
+++ b/aws_requests_auth/tests/test_boto_utils.py
@@ -2,8 +2,7 @@
 import hashlib
 import os
 import unittest
-
-import mock
+from unittest import mock
 
 from aws_requests_auth.boto_utils import BotoAWSRequestsAuth, get_credentials
 
