diff --git a/packages/aws-durable-execution-sdk-python-insight/src/aws_durable_execution_sdk_python_insight/plugin.py b/packages/aws-durable-execution-sdk-python-insight/src/aws_durable_execution_sdk_python_insight/plugin.py
index be1a93c8..ac6e7dae 100644
--- a/packages/aws-durable-execution-sdk-python-insight/src/aws_durable_execution_sdk_python_insight/plugin.py
+++ b/packages/aws-durable-execution-sdk-python-insight/src/aws_durable_execution_sdk_python_insight/plugin.py
@@ -342,9 +342,9 @@ def on_invocation_end(self, info: InvocationEndInfo) -> None:
         if should_emit:
             # Only terminal (SUCCEEDED/FAILED) records carry an end time; a
             # PENDING/RETRY invocation end maps to RUNNING (still in flight) and
-            # must omit endTime/durationMs. Passing end_time=None makes _emit
-            # drop both fields. Output and error likewise belong only to a
-            # terminal record.
+            # must omit endTime/durationMs. Passing end_time=None makes
+            # _schedule_record drop both fields. Output and error likewise
+            # belong only to a terminal record.
             self._schedule_record(
                 arn,
                 state,