diff --git a/tests/e2e_test/test_migration.py b/tests/e2e_test/test_migration.py
index 31fd5128..88043ea2 100644
--- a/tests/e2e_test/test_migration.py
+++ b/tests/e2e_test/test_migration.py
@@ -116,7 +116,7 @@ def run_bench_command(command):
         bench_command = generate_bench_command(
             ip_ports=f"127.0.0.1:{base_port + i}",
             model=model,
-            num_prompts=500,
+            num_prompts=300,
             dataset_type="sharegpt",
             dataset_path="/mnt/dataset/sharegpt_gpt4/sharegpt_gpt4.jsonl",
             qps=10,