diff --git a/agent.py b/agent.py
index ebe12e3..efff621 100644
--- a/agent.py
+++ b/agent.py
@@ -14,7 +14,7 @@ from pathlib import Path
 from typing import Optional
 from zoneinfo import ZoneInfo
 
-from claude_code_sdk import ClaudeSDKClient
+from claude_agent_sdk import ClaudeSDKClient
 
 # Fix Windows console encoding for Unicode characters (emoji, etc.)
 # Without this, print() crashes when Claude outputs emoji like ✅
diff --git a/ui/src/hooks/useWebSocket.ts b/ui/src/hooks/useWebSocket.ts
index cec2bf7..95ad084 100644
--- a/ui/src/hooks/useWebSocket.ts
+++ b/ui/src/hooks/useWebSocket.ts
@@ -112,6 +112,8 @@ export function useProjectWebSocket(projectName: string | null) {
               setState(prev => ({
                 ...prev,
                 agentStatus: message.status,
+                // Clear active agents when process stops to prevent stale UI
+                ...(message.status === 'stopped' && { activeAgents: [], recentActivity: [] }),
               }))
               break