diff --git a/client/driver/fabfile.py b/client/driver/fabfile.py
index 06e264a..cca9aff 100644
--- a/client/driver/fabfile.py
+++ b/client/driver/fabfile.py
@@ -1033,7 +1033,7 @@ def task_status_ui_test():
 
 def simulate_db_run(i, next_conf):
     # Using 1 knob to decide performance; simple but effective
-    gain = int(next_conf['effective_cache_size'].replace('kB', ''))
+    gain = int(next_conf['effective_cache_size'].replace('MB', '000').replace('kB', ''))
 
     with open('./integrationTests/data/x__metrics_after.json', 'r') as fin:
         metrics_after = json.load(fin)