diff --git a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py
index e32f556503de840c4fe49627866c80ea33629922..b4c7680214b38038576a7f8b0940a8e41c74fb73 100644
--- a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py
+++ b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py
@@ -369,7 +369,7 @@ class RunGenerator(object):
                 self.name_idx = idx
             elif column == 'Duration(us)':
                 self.duration_idx = idx
-            elif column == 'Type':
+            elif column == 'Accelerator Core':
                 self.core_type_idx = idx
 
             if column in display_columns: