diff --git a/tests/test_strategy.py b/tests/test_strategy.py
index bdcc2fa..5d22ab2 100644
--- a/tests/test_strategy.py
+++ b/tests/test_strategy.py
@@ -109,7 +109,8 @@ class TestStrategy(unittest.TestCase):
         }
         load = (0.7 * max(node_loads_08.values()) +
                 0.3 * max(node_loads_05.values()))
-        cap = 1 / load
+        cap = (0.7 * 1 / max(node_loads_08.values()) +
+                0.3 * 1/ max(node_loads_05.values()))
         self.assertAlmostEqual(sigma.load(read_fraction=fr), load)
         self.assertAlmostEqual(sigma.capacity(read_fraction=fr), cap)
         node_throughputs = {