diff --git a/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideField.vue b/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideField.vue
index 1ebcf4f0ef..a20973bc3c 100644
--- a/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideField.vue
+++ b/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideField.vue
@@ -58,7 +58,7 @@ const emit = defineEmits<{
 }
 
 .overrideCloseButton {
-	padding: 0px 8px 3px; // the icon used is off-center vertically
+	padding: 0px var(--spacing-2xs);
 	border: 0px;
 	color: var(--color-text-base);
 	margin-left: auto;
@@ -68,6 +68,7 @@ const emit = defineEmits<{
 
 .contentOverrideContainer {
 	display: flex;
+	align-items: center;
 	white-space: nowrap;
 	width: 100%;
 	gap: var(--spacing-4xs);