diff --git a/src/aiu_trace_analyzer/pipeline/iteration_detect.py b/src/aiu_trace_analyzer/pipeline/iteration_detect.py
index 967ca97..c864c6a 100644
--- a/src/aiu_trace_analyzer/pipeline/iteration_detect.py
+++ b/src/aiu_trace_analyzer/pipeline/iteration_detect.py
@@ -6,6 +6,7 @@
 import aiu_trace_analyzer.logger as aiulog
 from aiu_trace_analyzer.types import TraceEvent
 from aiu_trace_analyzer.pipeline import AbstractContext, AbstractHashQueueContext
+from aiu_trace_analyzer.pipeline.tools import PipelineContextTool
 
 
 class IterationStatus(object):
@@ -149,7 +150,7 @@ def drain(self) -> list[TraceEvent]:
 def collect_iteration_stats(event: TraceEvent, context: AbstractContext) -> list[TraceEvent]:
     assert isinstance(context, IterationDectectContext)
 
-    if event["ph"] in "X" and "args" in event and "TS1" in event["args"] and "Cmpt Exec" in event["name"]:
+    if event["ph"] == "X" and PipelineContextTool.is_acc_kernel(event):
         context.detect_iteration(event)
 
     return [event]