diff --git a/app/components/chat.tsx b/app/components/chat.tsx
index 25c3be26..ff0bc5b3 100644
--- a/app/components/chat.tsx
+++ b/app/components/chat.tsx
@@ -922,7 +922,7 @@ export function Chat() {
                         }}
                       ></IconButton>
                     </div>
-                    {message.role === "user" ? (
+                    {isUser ? (
                       <Avatar avatar={config.avatar} />
                     ) : (
                       <MaskAvatar mask={session.mask} />