diff --git a/profiler/cluster_analyse/communication_group/communication_group_generator.py b/profiler/cluster_analyse/communication_group/communication_group_generator.py
index 0202ea0e56da40f007d07d0e1149c7c460ebecd7..6dabcac3b6463196cd3c7f47c13c107536655fd0 100644
--- a/profiler/cluster_analyse/communication_group/communication_group_generator.py
+++ b/profiler/cluster_analyse/communication_group/communication_group_generator.py
@@ -69,7 +69,7 @@ class CommunicationGroupGenerator:
     def generate_p2p_communication_group(self):
         stage_group = {}
         for rank_set in self.collective_group_dict.values():
-            unioned_set = {}
+            unioned_set = set()
             remove_key = []
             for first_rank, stage in stage_group.items():
                 if UnionFind.is_connected(rank_set, stage):
diff --git a/profiler/cluster_analyse/prof_bean/step_trace_time_bean.py b/profiler/cluster_analyse/prof_bean/step_trace_time_bean.py
index 80f09d5615ae3777b0a7c2f7edd00864595de23b..b0a3be4f5eaccea70aa912bc85e68d70dbda3bde 100644
--- a/profiler/cluster_analyse/prof_bean/step_trace_time_bean.py
+++ b/profiler/cluster_analyse/prof_bean/step_trace_time_bean.py
@@ -31,8 +31,8 @@ class StepTraceTimeBean:
         return row
 
     @property
-    def step(self) -> float:
-        return float(self._data.get(self.STEP, ''))
+    def step(self) -> str:
+        return self._data.get(self.STEP, '')
 
     @property
     def all_headers(self) -> list: