Fix benchmark_throughput args

1 files changed, 1 insertions, 1 deletions

diff --git a/benchmark_throughput.py b/benchmark_throughput.py
index 5bfa4d5..8a9d4a5 100644
--- a/benchmark_throughput.py
+++ b/benchmark_throughput.py
@@ -50,7 +50,7 @@ def run_benchmark(duration_seconds: int, payload_size: int = 100):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="FemtoQueue throughput benchmark")
-    parser.add_argument("duration", type=int, help="Duration to run the benchmark (in seconds)", default=10)
+    parser.add_argument("--duration", type=int, help="Duration to run the benchmark (in seconds)", default=10, required=False)
     args = parser.parse_args()
 
     run_benchmark(args.duration)