diff --git a/src/components/xt-chat/chat-view/useChatHandler.tsx b/src/components/xt-chat/chat-view/useChatHandler.tsx
index 4d2f33b..b000a47 100644
--- a/src/components/xt-chat/chat-view/useChatHandler.tsx
+++ b/src/components/xt-chat/chat-view/useChatHandler.tsx
@@ -237,9 +237,11 @@ export default function useChatHandler({ initSse }): UseChatHandlerReturn {
onCopy(existingItem.content.output)}>
-
-
-
+ {team_session_state && (
+
+
+
+ )}
{isLastTask(data) && (
onRefresh(run_id)}>