diff --git a/tests/test_progressbar.py b/tests/test_progressbar.py
index 10ae74f..1ef2971 100644
--- a/tests/test_progressbar.py
+++ b/tests/test_progressbar.py
@@ -1,8 +1,8 @@
 # the inclusion of the tests module is not meant to offer best practices for
 # testing in general, but rather to support the `find_packages` example in
 # setup.py that excludes installing the 'tests' package
-# from mock import patch, Mock, MagicMock
-from unittest import TestCase, mock
+from unittest.mock import patch, Mock, MagicMock
+from unittest import TestCase
 from console_progressbar import ProgressBar