diff --git a/components/ChatWindow.tsx b/components/ChatWindow.tsx
index 794a34cf3..63668cdd2 100644
--- a/components/ChatWindow.tsx
+++ b/components/ChatWindow.tsx
@@ -486,7 +486,9 @@ export function ChatWindow({ session, newSessionCwd, onAgentEnd, onSessionCreate
+
{chatInputElement}
+
) : (
@@ -510,8 +512,6 @@ export function ChatWindow({ session, newSessionCwd, onAgentEnd, onSessionCreate
-
-
{(() => {
const toolResultsMap = new Map
();
for (const msg of messages) {
@@ -746,10 +746,20 @@ export function ChatWindow({ session, newSessionCwd, onAgentEnd, onSessionCreate
}}
>
-
+
{chatInputElement}
+
>