diff --git a/python/sglang/bench_one_batch.py b/python/sglang/benchmark/one_batch.py
similarity index 100%
rename from python/sglang/bench_one_batch.py
rename to python/sglang/benchmark/one_batch.py
diff --git a/python/sglang/test/test_utils.py b/python/sglang/test/test_utils.py
index 34d2d1e56..d2f898364 100644
--- a/python/sglang/test/test_utils.py
+++ b/python/sglang/test/test_utils.py
@@ -1564,7 +1564,7 @@ def run_bench_one_batch(model, other_args):
     command = [
         "python3",
         "-m",
-        "sglang.bench_one_batch",
+        "sglang.benchmark.one_batch",
         "--batch-size",
         "1",
         "--input",
diff --git a/test/manual/test_forward_split_prefill.py b/test/manual/test_forward_split_prefill.py
index f48ef6b0a..3ca254458 100644
--- a/test/manual/test_forward_split_prefill.py
+++ b/test/manual/test_forward_split_prefill.py
@@ -13,7 +13,7 @@ from array import array
 import numpy as np
 import torch
 
-from sglang.bench_one_batch import TreeCacheNamespace
+from sglang.benchmark.one_batch import TreeCacheNamespace
 from sglang.srt.configs.model_config import ModelConfig
 from sglang.srt.managers.schedule_batch import Req, ScheduleBatch
 from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode