diff --git a/src/cli/run/completion-continuation.test.ts b/src/cli/run/completion-continuation.test.ts
index 795a32ff0..976277cca 100644
--- a/src/cli/run/completion-continuation.test.ts
+++ b/src/cli/run/completion-continuation.test.ts
@@ -119,6 +119,7 @@ describe("checkCompletionConditions continuation coverage", () => {
 
     const ctx = createMockContext(directory)
     ctx.sessionID = "child-session"
+    setSessionAgent("child-session", "atlas")
     ctx.client.session.get = mock(async ({ path }: { path: { id: string } }) => ({
       data: {
         id: path.id,
@@ -400,6 +401,7 @@ describe("checkCompletionConditions continuation coverage", () => {
 
     const ctx = createMockContext(directory)
     ctx.sessionID = "ses_child_after_compaction"
+    setSessionAgent("ses_child_after_compaction", "atlas")
     ctx.client.session.get = mock(async ({ path }: { path: { id: string } }) => ({
       data: {
         id: path.id,