diff --git a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor
index 52999549..652dac14 100644
--- a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor	
+++ b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor	
@@ -105,10 +105,14 @@
                                         {
                                             <MudMarkdown @key="@segment.RenderKey" Value="@segmentContent" Props="Markdown.DefaultConfig" Styling="@this.MarkdownStyling" MarkdownPipeline="Markdown.CHAT_MARKDOWN_PIPELINE" />
                                         }
-                                        else
+                                        else if (segment.Type is MarkdownRenderSegmentType.MATH_BLOCK)
                                         {
                                             <MathJaxBlock @key="@segment.RenderKey" Value="@segmentContent" Class="mb-5" />
                                         }
+                                        else if (segment.ChartResult is not null)
+                                        {
+                                            <ChartBlock @key="@segment.RenderKey" Result="@segment.ChartResult" />
+                                        }
                                     }
                                     @if (textContent.Sources.Count > 0)
                                     {