diff --git a/x-pack/plugins/aiops/public/components/log_categorization/log_categorization_for_embeddable/discover_tabs.tsx b/x-pack/plugins/aiops/public/components/log_categorization/log_categorization_for_embeddable/discover_tabs.tsx
index 96fae405eec2c..eede82d50076d 100644
--- a/x-pack/plugins/aiops/public/components/log_categorization/log_categorization_for_embeddable/discover_tabs.tsx
+++ b/x-pack/plugins/aiops/public/components/log_categorization/log_categorization_for_embeddable/discover_tabs.tsx
@@ -62,7 +62,7 @@ export const DiscoverTabs: FC<Props> = ({
   query,
 }) => {
   return (
-    <EuiFlexItem grow={false}>
+    <EuiFlexItem grow={false} className="unifiedDataTableToolbar">
       <EuiFlexGroup gutterSize="none">
         <EuiFlexItem grow={false}>{renderViewModeToggle(data?.categories.length)}</EuiFlexItem>
         <EuiFlexItem />