diff --git a/packages/ui/src/components/message-part.tsx b/packages/ui/src/components/message-part.tsx
index c4a83077..beba90b4 100644
--- a/packages/ui/src/components/message-part.tsx
+++ b/packages/ui/src/components/message-part.tsx
@@ -277,6 +277,7 @@ export default function MessagePart(props: MessagePartProps) {
             toolCallId={props.part?.id}
             instanceId={props.instanceId}
             sessionId={props.sessionId}
+            onContentRendered={props.onRendered}
           />
         </Suspense>
       </Match>