diff --git a/tests/modules/test_traffic.py b/tests/modules/test_traffic.py
index 951a4d8..c312d97 100644
--- a/tests/modules/test_traffic.py
+++ b/tests/modules/test_traffic.py
@@ -14,10 +14,10 @@ class TestTrafficModule(unittest.TestCase):
 
     def test_get_minwidth_str(self):
         # default value (two digits after dot)
-        self.assertEqual(self.module.get_minwidth_str(), "1000.00MB")
+        self.assertEqual(self.module.get_minwidth_str(), "1000.00KiB/s")
         # integer value
         self.module._format = "{:.0f}"
-        self.assertEqual(self.module.get_minwidth_str(), "1000MB")
+        self.assertEqual(self.module.get_minwidth_str(), "1000KiB/s")
         # just one digit after dot
         self.module._format = "{:.1f}"
-        self.assertEqual(self.module.get_minwidth_str(), "1000.0MB")
+        self.assertEqual(self.module.get_minwidth_str(), "1000.0KiB/s")