diff --git a/tests/test_progressbar.py b/tests/test_progressbar.py
index 3503182..9b56d2e 100644
--- a/tests/test_progressbar.py
+++ b/tests/test_progressbar.py
@@ -6,7 +6,7 @@
 from console_progressbar import ProgressBar
 
 
-class TestConsoleProgressBar(unittest.TestCase):
+class TestConsoleProgressBar(TestCase):
     def setUp(self):
         self.total = 100