diff --git a/tests/correios_api_test.py b/tests/correios_api_test.py
index 9f68a14..6777ec3 100644
--- a/tests/correios_api_test.py
+++ b/tests/correios_api_test.py
@@ -1,6 +1,7 @@
 import unittest
 
-from mockito import when, Mock
+from mock import Mock
+from mockito import when
 
 from packtrack import Correios
 
diff --git a/tests/correios_test.py b/tests/correios_test.py
index ed27249..d24712e 100644
--- a/tests/correios_test.py
+++ b/tests/correios_test.py
@@ -1,5 +1,6 @@
 import unittest
 
+from mock import Mock
 from mockito import *
 
 from packtrack.correios import Encomenda, Status, EncomendaRepository