diff --git a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
index 90e406f500ba8..0731d06a5dd26 100644
--- a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
+++ b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
@@ -8,7 +8,7 @@ import { EuiCodeBlock, EuiFlexGroup, EuiFlexItem, EuiPanel, useEuiTheme } from '
 import { css } from '@emotion/css';
 import React from 'react';
 
-export const CustomCodeBlock = ({ value }: { value: string }) => {
+export const CustomCodeBlock = ({ value, lang }: { value: string; lang: string }) => {
   const theme = useEuiTheme();
 
   return (
@@ -31,7 +31,7 @@ export const CustomCodeBlock = ({ value }: { value: string }) => {
     >
       <EuiFlexGroup direction="column" gutterSize="xs">
         <EuiFlexItem grow={false}>
-          <EuiCodeBlock isCopyable fontSize="m">
+          <EuiCodeBlock isCopyable fontSize="m" language={lang}>
             {value}
           </EuiCodeBlock>
         </EuiFlexItem>
diff --git a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
index de7697f1bad8a..4095fc462ee01 100644
--- a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
+++ b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
@@ -132,7 +132,7 @@ const getPluginDependencies = ({
     customCodeBlock: (props) => {
       return (
         <>
-          <CustomCodeBlock value={props.value} />
+          <CustomCodeBlock value={props.value} lang={props.lang} />
           <EuiSpacer size="m" />
         </>
       );