diff --git a/css/src/ai-components/button.module.css b/css/src/ai-components/button.module.css
index 9c40e27162..ba95fa5764 100644
--- a/css/src/ai-components/button.module.css
+++ b/css/src/ai-components/button.module.css
@@ -48,8 +48,8 @@
 }
 
 .AIButton:focus {
-  outline: 0;
-  box-shadow: var(--shadow-spread) var(--primary-shadow);
+  outline: var(--border-width-05) solid var(--primary-focus);
+  outline-offset: var(--spacing-05);
 }
 
 /* Button Icon */
diff --git a/css/src/ai-components/iconButton.module.css b/css/src/ai-components/iconButton.module.css
index 747a0410ad..b3823bf676 100644
--- a/css/src/ai-components/iconButton.module.css
+++ b/css/src/ai-components/iconButton.module.css
@@ -39,10 +39,9 @@
   color: var(--secondary-dark) !important;
 }
 
-.AIIconButton:focus,
-.AIIconButton:focus-visible {
-  box-shadow: var(--shadow-spread) var(--secondary-shadow);
-  outline: none;
+.AIIconButton:focus {
+  outline: var(--border-width-05) solid var(--primary-focus);
+  outline-offset: var(--spacing-05);
 }
 
 .AIIconButton:disabled {