diff --git a/tests/test_partitions.py b/tests/test_partitions.py
index 4b7d32c1d68f634b6793fe4925ec92e5a333b623..9be8596088d372bfe82743f1cebf49b52dfe9e5a 100755
--- a/tests/test_partitions.py
+++ b/tests/test_partitions.py
@@ -92,12 +92,14 @@ def check_allocation(dev):
         size = int(blocks) * 512
         if name == root_dev:
             max_size = size
+            if root_dev.startswith('md'):
+                total_size += size
         else:
             total_size += size
     unallocated = max_size - total_size
     unallocated_gbytes = to_gbytes(unallocated)
     if unallocated_gbytes > 1:
-        print('Warning, %s GB are unallocated on %s.' % (unallocated_gbytes, root_dev))
+        print('Warning: %s%s GB are unallocated on %s.%s' % (YELLOW, unallocated_gbytes, root_dev, DEF))
         return False
     return True