diff --git a/packages/server/src/server/routes/storage.ts b/packages/server/src/server/routes/storage.ts
index 60e492e..a12f37a 100644
--- a/packages/server/src/server/routes/storage.ts
+++ b/packages/server/src/server/routes/storage.ts
@@ -3,7 +3,7 @@ import { z } from "zod"
 import { InstanceStore } from "../../storage/instance-store"
 import { EventBus } from "../../events/bus"
 import { ModelPreferenceSchema } from "../../config/schema"
-import type { InstanceData, Task, SessionTasks } from "../../api-types"
+import type { InstanceData } from "../../api-types"
 import { WorkspaceManager } from "../../workspaces/manager"
 
 interface RouteDeps {
@@ -18,6 +18,8 @@ const TaskSchema = z.object({
   status: z.enum(["completed", "interrupted", "in-progress", "pending"]),
   timestamp: z.number(),
   messageIds: z.array(z.string()).optional(),
+  taskSessionId: z.string().optional(),
+  archived: z.boolean().optional(),
 })
 
 const InstanceDataSchema = z.object({
diff --git a/packages/ui/src/components/settings/AntigravitySettings.tsx b/packages/ui/src/components/settings/AntigravitySettings.tsx
index 852a9a8..679ad62 100644
--- a/packages/ui/src/components/settings/AntigravitySettings.tsx
+++ b/packages/ui/src/components/settings/AntigravitySettings.tsx
@@ -126,7 +126,8 @@ const AntigravitySettings: Component = () => {
             })
 
             if (!response.ok) {
-                throw new Error('Failed to start authentication')
+                const errorData = await response.json().catch(() => ({}))
+                throw new Error(errorData.error || errorData.details || 'Failed to start authentication')
             }
 
             const data = await response.json() as {