diff --git a/main.py b/main.py
index 776f42b28..b442c39f7 100755
--- a/main.py
+++ b/main.py
@@ -755,9 +755,14 @@ def mlperf_inference_run_command(
             if "dlrm-v2" in model and implementation == "nvidia":
                 cmd_suffix += f" \\\n{pre_space} --criteo_day23_raw_data_path=<PATH_TO_CRITEO_DAY23_RAW_DATA>"
 
+            if "short" in extra_variation_tags:
+                full_ds_needed_tag = ""
+            else:
+                full_ds_needed_tag = "_full,"
+
             run_cmd = f"""\n
 {f_pre_space}```bash
-{f_pre_space}mlcr run-mlperf,inference,_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
+{f_pre_space}mlcr run-mlperf,inference,{full_ds_needed_tag}_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
 {pre_space} --model={model} \\
 {pre_space} --implementation={implementation} \\
 {pre_space} --framework={framework} \\