diff --git a/tests/test_progressbar.py b/tests/test_progressbar.py
index 52888cf..10ae74f 100644
--- a/tests/test_progressbar.py
+++ b/tests/test_progressbar.py
@@ -2,7 +2,7 @@
 # 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
+from unittest import TestCase, mock
 from console_progressbar import ProgressBar