From 972f810ac72028187cb4d3557b5737b2d3a5710a Mon Sep 17 00:00:00 2001
From: troymc <troy@ascribe.io>
Date: Wed, 11 May 2016 15:43:30 +0200
Subject: [PATCH] get payload size choices from SIZE_OF_FILLER.keys()

---
 benchmarking-tests/benchmark_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarking-tests/benchmark_utils.py b/benchmarking-tests/benchmark_utils.py
index 724dc43e..fa50bfa9 100644
--- a/benchmarking-tests/benchmark_utils.py
+++ b/benchmarking-tests/benchmark_utils.py
@@ -119,7 +119,7 @@ def main():
                                 type=int, default=0,
                                 help='Number of transactions to add to the backlog')
     backlog_parser.add_argument('-s', '--payload-size',
-                                choices=['minimal', 'small', 'medium', 'large'],
+                                choices=SIZE_OF_FILLER.keys(),
                                 default='minimal',
                                 help='Payload size')