diff --git a/app/ui/app/src/components/StreamingMarkdownContent.tsx b/app/ui/app/src/components/StreamingMarkdownContent.tsx
index 5f97a39774..249a203a81 100644
--- a/app/ui/app/src/components/StreamingMarkdownContent.tsx
+++ b/app/ui/app/src/components/StreamingMarkdownContent.tsx
@@ -211,7 +211,7 @@ const StreamingMarkdownContent: React.FC<StreamingMarkdownContentProps> =
             parseIncompleteMarkdown={isStreaming}
             isAnimating={isStreaming}
             remarkPlugins={remarkPlugins}
-            disableTableActions={true}
+            controls={false}
             components={{
               pre: CodeBlock,
               table: ({
diff --git a/app/ui/app/src/index.css b/app/ui/app/src/index.css
index f1768b3feb..af1f265dc4 100644
--- a/app/ui/app/src/index.css
+++ b/app/ui/app/src/index.css
@@ -28,17 +28,3 @@
     opacity: 1;
   }
 }
-
-/* Hide Streamdown table action buttons */
-.prose button[title="Copy table as markdown"],
-.prose button[title="Download table"] {
-  display: none !important;
-}
-
-/* Hide the parent div if it only contains these buttons */
-.prose
-  div:has(> button[title="Copy table as markdown"]):has(
-    > button[title="Download table"]
-  ) {
-  display: none !important;
-}