diff --git a/debug/accuracy_tools/msprobe/visualization/utils.py b/debug/accuracy_tools/msprobe/visualization/utils.py
index 15c7052f79afc56306e5f64ac475a4a7a2a5fa87..b86b738f2ca7412930991c376858375b23bb25bb 100644
--- a/debug/accuracy_tools/msprobe/visualization/utils.py
+++ b/debug/accuracy_tools/msprobe/visualization/utils.py
@@ -131,7 +131,7 @@ def load_parallel_param(input_param):
     config_b = parallel_merge.get('bench', {})
     return (ParallelParam(config_n.get('rank_size'), config_n.get('tp'), config_n.get('pp')),) if not config_b else \
         (ParallelParam(config_n.get('rank_size'), config_n.get('tp'), config_n.get('pp'), config_n.get('vpp', 1)),
-         ParallelParam(config_b.get('rank_size'), config_b.get('tp'), config_b.get('pp'), config_n.get('vpp', 1)))
+         ParallelParam(config_b.get('rank_size'), config_b.get('tp'), config_b.get('pp'), config_b.get('vpp', 1)))
 
 
 def validate_parallel_param(parallel_param, dump_path, log_prefix='[NPU]'):