diff --git a/public/src/uncloseai-modal-builtin.css b/public/src/uncloseai-modal-builtin.css
index 4c8b818..8edab3f 100644
--- a/public/src/uncloseai-modal-builtin.css
+++ b/public/src/uncloseai-modal-builtin.css
@@ -932,7 +932,8 @@ dialog#translate-modal button {
 
 /* API Configuration Styles */
 .uncloseai-toggle-container {
-  display: flex;
+  display: grid;
+  grid-template-columns: 18px 1fr;
   align-items: center;
   gap: 8px;
   margin-bottom: 12px;
@@ -941,7 +942,6 @@ dialog#translate-modal button {
 .uncloseai-toggle-container input[type="checkbox"] {
   width: 18px;
   height: 18px;
-  flex-shrink: 0;
   cursor: pointer;
 }
 
@@ -1280,10 +1280,10 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .user-message {
   max-width: 280px;
   backdrop-filter: blur(4px);
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.4);
-  display: flex;
+  display: grid;
+  grid-auto-flow: column;
   align-items: center;
   gap: 8px;
-  flex-wrap: wrap;
   animation: tooltip-fade-in 0.3s ease-out;
 }
 
@@ -1508,9 +1508,8 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .user-message {
   position: absolute;
   inset: 0;
   z-index: 9999;
-  display: flex;
-  align-items: center;
-  justify-content: center;
+  display: grid;
+  place-items: center;
   padding: 24px;
   background: #1a1a2e;
 }
diff --git a/public/src/uncloseai-modal-pico.css b/public/src/uncloseai-modal-pico.css
index 159adfc..25f5636 100644
--- a/public/src/uncloseai-modal-pico.css
+++ b/public/src/uncloseai-modal-pico.css
@@ -1002,7 +1002,8 @@ dialog#translate-modal button {
 
 /* API Configuration Styles */
 .uncloseai-toggle-container {
-  display: flex;
+  display: grid;
+  grid-template-columns: 18px 1fr;
   align-items: center;
   gap: 8px;
   margin-bottom: 12px;
@@ -1011,7 +1012,6 @@ dialog#translate-modal button {
 .uncloseai-toggle-container input[type="checkbox"] {
   width: 18px;
   height: 18px;
-  flex-shrink: 0;
   cursor: pointer;
 }
 
@@ -1358,10 +1358,10 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .user-message {
   max-width: 280px !important;
   backdrop-filter: blur(4px) !important;
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.4) !important;
-  display: flex !important;
+  display: grid !important;
+  grid-auto-flow: column !important;
   align-items: center !important;
   gap: 8px !important;
-  flex-wrap: wrap !important;
   animation: tooltip-fade-in 0.3s ease-out !important;
 }
 
@@ -1568,9 +1568,8 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .user-message {
   position: absolute;
   inset: 0;
   z-index: 9999;
-  display: flex;
-  align-items: center;
-  justify-content: center;
+  display: grid;
+  place-items: center;
   padding: 24px;
   background: #1a1a2e;
 }