diff --git a/src/Api/NtbPrompt.ts b/src/Api/NtbPrompt.ts
index 2f8d7ab..69e6f00 100644
--- a/src/Api/NtbPrompt.ts
+++ b/src/Api/NtbPrompt.ts
@@ -48,7 +48,7 @@ export class NtbPrompt extends Modal {
         this.placeholder = placeholder;
         this.default = default_value;
 
-        this.modalEl.addClasses(['prompt', 'note-toolbar-ui-modal']);
+        this.modalEl.addClasses(['prompt', 'note-toolbar-ui']);
         if (!this.label) this.modalEl.setAttr('data-ntb-ui-mode', 'compact');
     }
 
diff --git a/src/Api/NtbSuggester.ts b/src/Api/NtbSuggester.ts
index ea4aa6b..56ab3cb 100644
--- a/src/Api/NtbSuggester.ts
+++ b/src/Api/NtbSuggester.ts
@@ -34,7 +34,7 @@ export class NtbSuggester<T> extends FuzzySuggestModal<T> {
         } = options || {};
 
         this.setPlaceholder(placeholder ? placeholder : t('api.ui.suggester-placeholder'));
-        this.modalEl.addClass("note-toolbar-ui-modal");
+        this.modalEl.addClass("note-toolbar-ui");
         if (!keys) {
             if (Array.isArray(values)) {
                 // if it's a string array, convert it to T[] (if possible)
diff --git a/src/styles.css b/src/styles.css
index 2d2ad89..4c7e347 100644
--- a/src/styles.css
+++ b/src/styles.css
@@ -1741,27 +1741,7 @@ input[type="text"].note-toolbar-setting-error,
  * Components (Note Toolbar API)
  *************************************************************************/
 
-.note-toolbar-ui-button-div {
-    display: flex;
-    flex-direction: column;
-    align-items: end;
-    margin-top: 1rem;
-}
-
-.note-toolbar-ui-div {
-    display: flex;
-}
-
-.note-toolbar-ui-form {
-    display: flex;
-    flex-grow: 1;
-}
-
-.note-toolbar-ui-input {
-    flex-grow: 1;
-}
-
-.note-toolbar-ui-modal {
+ .note-toolbar-ui {
     min-height: unset;
     &.prompt {
         .is-mobile & {
@@ -1781,4 +1761,24 @@ input[type="text"].note-toolbar-setting-error,
             display: none;
         }
     }
+}
+
+.note-toolbar-ui-button-div {
+    display: flex;
+    flex-direction: column;
+    align-items: end;
+    margin-top: 1rem;
+}
+
+.note-toolbar-ui-div {
+    display: flex;
+}
+
+.note-toolbar-ui-form {
+    display: flex;
+    flex-grow: 1;
+}
+
+.note-toolbar-ui-input {
+    flex-grow: 1;
 }
\ No newline at end of file
diff --git a/styles.css b/styles.css
index 294292e..2cd5c2c 100644
--- a/styles.css
+++ b/styles.css
@@ -2630,27 +2630,7 @@ input[type="text"].note-toolbar-setting-error,
  * Components (Note Toolbar API)
  *************************************************************************/
 
-.note-toolbar-ui-button-div {
-    display: flex;
-    flex-direction: column;
-    align-items: end;
-    margin-top: 1rem;
-}
-
-.note-toolbar-ui-div {
-    display: flex;
-}
-
-.note-toolbar-ui-form {
-    display: flex;
-    flex-grow: 1;
-}
-
-.note-toolbar-ui-input {
-    flex-grow: 1;
-}
-
-.note-toolbar-ui-modal {
+ .note-toolbar-ui {
     min-height: unset;
     &.prompt {
         .is-mobile & {
@@ -2670,4 +2650,24 @@ input[type="text"].note-toolbar-setting-error,
             display: none;
         }
     }
+}
+
+.note-toolbar-ui-button-div {
+    display: flex;
+    flex-direction: column;
+    align-items: end;
+    margin-top: 1rem;
+}
+
+.note-toolbar-ui-div {
+    display: flex;
+}
+
+.note-toolbar-ui-form {
+    display: flex;
+    flex-grow: 1;
+}
+
+.note-toolbar-ui-input {
+    flex-grow: 1;
 }
\ No newline at end of file