diff --git a/packages/cli/src/ui/utils/historyExportUtils.ts b/packages/cli/src/ui/utils/historyExportUtils.ts
index a89c31c652..91eed28735 100644
--- a/packages/cli/src/ui/utils/historyExportUtils.ts
+++ b/packages/cli/src/ui/utils/historyExportUtils.ts
@@ -81,7 +81,7 @@ export async function exportHistoryToFile(
   const {
     messages,
     filePath,
-    trajectories: _trajectories, // Collected but not yet included in Stage 1 JSON output
+    trajectories: _trajectories, // Collected but not yet included in Stage 2 JSON output
     history: providedHistory,
   } = options;
   const extension = path.extname(filePath).toLowerCase();