diff --git a/x-pack/plugins/apm/public/components/shared/ml_callout/index.tsx b/x-pack/plugins/apm/public/components/shared/ml_callout/index.tsx
index 7ae0440bbef3a..12b4e021e2302 100644
--- a/x-pack/plugins/apm/public/components/shared/ml_callout/index.tsx
+++ b/x-pack/plugins/apm/public/components/shared/ml_callout/index.tsx
@@ -9,7 +9,7 @@ import {
   EuiButton,
   EuiButtonEmpty,
   EuiCallOut,
-  EuiFlexGrid,
+  EuiFlexGroup,
   EuiFlexItem,
 } from '@elastic/eui';
 import { i18n } from '@kbn/i18n';
@@ -167,7 +167,7 @@ export function MLCallout({
   const hasAnyActions = properties.primaryAction || dismissable;
 
   const actions = hasAnyActions ? (
-    <EuiFlexGrid gutterSize="s">
+    <EuiFlexGroup gutterSize="s" justifyContent="spaceBetween">
       {properties.primaryAction && (
         <EuiFlexItem grow={false}>{properties.primaryAction}</EuiFlexItem>
       )}
@@ -180,7 +180,7 @@ export function MLCallout({
           </EuiButtonEmpty>
         </EuiFlexItem>
       )}
-    </EuiFlexGrid>
+    </EuiFlexGroup>
   ) : null;
 
   return (