diff --git a/docs/AI-for-security/attack-discovery.asciidoc b/docs/AI-for-security/attack-discovery.asciidoc
index 7b0d32431a..01bceba437 100644
--- a/docs/AI-for-security/attack-discovery.asciidoc
+++ b/docs/AI-for-security/attack-discovery.asciidoc
@@ -56,7 +56,7 @@ It may take from a few seconds up to several minutes to generate discoveries, de
 
 IMPORTANT: Attack discovery is in technical preview and will only analyze opened and acknowleged alerts from the past 24 hours. By default it only analyzes up to 20 alerts within this timeframe, but you can expand this up to 100 by going to **AI Assistant → Settings (image:images/icon-settings.png[Settings icon,17,17]) → Knowledge Base** and updating the **Alerts** setting.
 
-image::images/knowledge-base-settings.png["AI Assistant's settings menu open to the Knowledge Base tab",75%]
+image::images/knowledge-base-assistant-settings-kb-tab.png["AI Assistant's settings menu open to the Knowledge Base tab",75%]
 
 IMPORTANT: Attack discovery uses the same data anonymization settings as <<security-assistant, Elastic AI Assistant>>. To configure which alert fields are sent to the LLM and which of those fields are obfuscated, use the Elastic AI Assistant settings. Consider the privacy policies of third-party LLMs before sending them sensitive data.
 
diff --git a/docs/AI-for-security/knowledge-base.asciidoc b/docs/AI-for-security/knowledge-base.asciidoc
index cd942726a5..2a157411e0 100644
--- a/docs/AI-for-security/knowledge-base.asciidoc
+++ b/docs/AI-for-security/knowledge-base.asciidoc
@@ -27,7 +27,7 @@ NOTE: You must individually enable knowledge base for each {kib} space where you
 
 Open a conversation with AI Assistant, select a large language model, then click **Setup Knowledge Base**. If the button doesn't appear, knowledge base is already enabled.
 
-image::images/knowledge-base-assistant-setup-button[An AI Assistant conversation showing the Setup Knowledge Base button]
+image::images/knowledge-base-assistant-setup-button.png[An AI Assistant conversation showing the Setup Knowledge Base button]
 
 Knowledge base setup may take several minutes. It will continue in the background if you close the conversation. After setup is complete, you can access knowledge base settings from the conversation settings menu.