diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/chat/chat_item_controls.tsx b/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/chat/chat_item_controls.tsx
index 73663416a861f..2b8487fa45c9d 100644
--- a/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/chat/chat_item_controls.tsx
+++ b/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/chat/chat_item_controls.tsx
@@ -64,7 +64,7 @@ export function ChatItemControls({
 
   return controls ? (
     <>
-      <EuiHorizontalRule margin="none" color={euiTheme.colors.backgroundBaseDisabled} />
+      <EuiHorizontalRule margin="none" color={euiTheme.colors.lightestShade} />
       <EuiPanel hasShadow={false} paddingSize="s" className={containerClassName}>
         {controls}
       </EuiPanel>
diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/message_panel/esql_code_block.tsx b/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/message_panel/esql_code_block.tsx
index 56547173df951..24347c3aadb5e 100644
--- a/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/message_panel/esql_code_block.tsx
+++ b/x-pack/plugins/observability_solution/observability_ai_assistant/public/components/message_panel/esql_code_block.tsx
@@ -19,7 +19,7 @@ import React from 'react';
 import { ChatActionClickHandler, ChatActionClickType } from '../chat/types';
 
 const getCodeBlockClassName = (theme: UseEuiTheme) => css`
-  background-color: ${theme.euiTheme.colors.backgroundBaseDisabled};
+  background-color: ${theme.euiTheme.colors.lightestShade};
   .euiCodeBlock__pre {
     margin-bottom: 0;
     padding: ${theme.euiTheme.size.m};