diff --git a/tools/llm_bench/llm_bench_utils/model_utils.py b/tools/llm_bench/llm_bench_utils/model_utils.py
index 56800cbea8..c21ba161ec 100644
--- a/tools/llm_bench/llm_bench_utils/model_utils.py
+++ b/tools/llm_bench/llm_bench_utils/model_utils.py
@@ -298,7 +298,7 @@ def get_model_name(model_name_or_path, task=None):
         for use_case in possible_use_cases:
             for m_type in use_case.model_types:
                 if model_name.lower().startswith(m_type):
-                    return use_case, m_type
+                    return use_case, model_name
 
     return None, None