diff --git a/dist/src/taskState.js b/dist/src/taskState.js
index 57f5269..e6f4c2a 100644
--- a/dist/src/taskState.js
+++ b/dist/src/taskState.js
@@ -176,7 +176,7 @@ export async function getXWorkmateTaskSnapshot(input) {
                 status: recordedRun.status,
                 taskStatus: recordedRun.status,
                 terminal: recordedRun.status !== "running",
-                terminalSource: "agent_end",
+                terminalSource: recordedRun.status === "running" ? "session_prepare" : "agent_end",
                 mode: "gateway-chat",
                 mapping,
                 appThreadKey: mapping?.appThreadKey ?? appThreadKey,
diff --git a/src/taskState.test.ts b/src/taskState.test.ts
index b955e6f..8fd0f67 100644
--- a/src/taskState.test.ts
+++ b/src/taskState.test.ts
@@ -330,6 +330,7 @@ describe("xworkmate task state mapping", () => {
       success: true,
       status: "running",
       terminal: false,
+      terminalSource: "session_prepare",
       task: {
         runId: "turn-running",
         status: "running",
diff --git a/src/taskState.ts b/src/taskState.ts
index b63542c..7381c3f 100644
--- a/src/taskState.ts
+++ b/src/taskState.ts
@@ -293,7 +293,7 @@ export async function getXWorkmateTaskSnapshot(input: {
         status: recordedRun.status,
         taskStatus: recordedRun.status,
         terminal: recordedRun.status !== "running",
-        terminalSource: "agent_end",
+        terminalSource: recordedRun.status === "running" ? "session_prepare" : "agent_end",
         mode: "gateway-chat",
         mapping,
         appThreadKey: mapping?.appThreadKey ?? appThreadKey,