diff --git a/packages/server/src/api-types.ts b/packages/server/src/api-types.ts
index 1703af7..adcc102 100644
--- a/packages/server/src/api-types.ts
+++ b/packages/server/src/api-types.ts
@@ -16,6 +16,8 @@ export interface Task {
   status: TaskStatus
   timestamp: number
   messageIds?: string[] // IDs of messages associated with this task
+  taskSessionId?: string
+  archived?: boolean
 }
 
 export interface SessionTasks {
diff --git a/packages/ui/src/components/chat/multix-v2/index.tsx b/packages/ui/src/components/chat/multix-v2/index.tsx
index 56a59ef..026e628 100644
--- a/packages/ui/src/components/chat/multix-v2/index.tsx
+++ b/packages/ui/src/components/chat/multix-v2/index.tsx
@@ -10,7 +10,7 @@
 import { createSignal, Show, onMount, For, onCleanup, batch } from "solid-js";
 import toast from "solid-toast";
 import { sessions, activeSessionId, setActiveSession } from "@/stores/session-state";
-import { sendMessage, compactSession, updateSessionAgent, updateSessionModelForSession, forceReset, abortSession } from "@/stores/session-actions";
+import { sendMessage, compactSession, updateSessionAgent, updateSessionModelForSession, forceReset, abortSession, loadMessages, fetchSessions } from "@/stores/session-actions";
 import { addTask, setActiveTask, archiveTask } from "@/stores/task-actions";
 import { messageStoreBus } from "@/stores/message-v2/bus";
 import { formatTokenTotal } from "@/lib/formatters";
@@ -216,6 +216,9 @@ export default function MultiXV2(props: MultiXV2Props) {
             setSendingTasks(new Set<string>());
         });
 
+        // Initialize
+        loadMessages(props.instanceId, props.sessionId);
+        fetchSessions(props.instanceId);
         syncFromStore();
 
         const interval = setInterval(syncFromStore, 150);