diff --git a/examples/benchmark.py b/examples/benchmark.py
index 331b1b0..af71037 100644
--- a/examples/benchmark.py
+++ b/examples/benchmark.py
@@ -560,7 +560,7 @@ def main(dp_size: int, dp_rank: int) -> None:
 
     if profiling_enabled:
         trace_stem = (
-            f"{args.model_name}-ep{ep_size}-cp{cp_size}"
+            f"{Path(args.model_name).name}-ep{ep_size}-cp{cp_size}"
             f"-prefillbsz{args.prefill_batch_size}-noncached{args.non_cached_tokens}"
             f"-input{args.context_length}-bs{args.decode_num_requests}"
             f"-nextn{envs.DEEPINFER_USE_NEXTN_DECODE}-force_topk{args.sampler_force_topk}"