diff --git a/pre-upload_unittest.py b/pre-upload_unittest.py
index 0702f54..a318242 100755
--- a/pre-upload_unittest.py
+++ b/pre-upload_unittest.py
@@ -12,8 +12,7 @@
 import datetime
 import os
 import sys
-
-import mock
+from unittest import mock
 
 import errors