diff --git a/packages/server/src/server/agent/import-sessions.test.ts b/packages/server/src/server/agent/import-sessions.test.ts
index 470b36430..7c6093bd1 100644
--- a/packages/server/src/server/agent/import-sessions.test.ts
+++ b/packages/server/src/server/agent/import-sessions.test.ts
@@ -374,6 +374,7 @@ test("importProviderSession imports a selected provider session without listing"
   const agentManager = {
     importProviderSession: vi.fn().mockResolvedValue(snapshot),
     getTimeline: vi.fn().mockReturnValue(timeline),
+    unarchiveSnapshot: vi.fn().mockResolvedValue(false),
   } as unknown as AgentManager;
   const agentStorage = {
     list: vi.fn().mockResolvedValue([]),
@@ -414,6 +415,7 @@ test("importProviderSession passes labels through the manager import operation",
   const agentManager = {
     importProviderSession: vi.fn().mockResolvedValue(snapshot),
     getTimeline: vi.fn().mockReturnValue([]),
+    unarchiveSnapshot: vi.fn().mockResolvedValue(false),
   } as unknown as AgentManager;
   const agentStorage = {
     list: vi.fn().mockResolvedValue([]),
diff --git a/packages/server/src/server/session.workspaces.test.ts b/packages/server/src/server/session.workspaces.test.ts
index 876825e5b..ce8639111 100644
--- a/packages/server/src/server/session.workspaces.test.ts
+++ b/packages/server/src/server/session.workspaces.test.ts
@@ -543,6 +543,7 @@ function createSessionForWorkspaceTests(
         getAgent: () => null,
         archiveAgent: async () => ({ archivedAt: new Date().toISOString() }),
         archiveSnapshot: async () => ({}),
+        unarchiveSnapshot: async () => true,
         clearAgentAttention: async () => {},
         notifyAgentState: () => {},
       }),