diff --git a/compaction_test.py b/compaction_test.py
index f7b27078fd..c5d90560d9 100644
--- a/compaction_test.py
+++ b/compaction_test.py
@@ -286,10 +286,11 @@ def test_compaction_throughput(self):
             "MB": 1,
             "MiB": 1,
             "KiB": 1. / 1024,
-            "GiB": 1024
+            "GiB": 1024,
+            "B": 1. / (1024 * 1024),
         }
 
-        units = ['MB'] if cluster.version() < LooseVersion('3.6') else ['KiB', 'MiB', 'GiB']
+        units = ['MB'] if cluster.version() < LooseVersion('3.6') else ['B', 'KiB', 'MiB', 'GiB']
         assert found_units in units
 
         logger.debug(avgthroughput)