diff --git a/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideButton.vue b/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideButton.vue
index 5867ff0d7b..5d69dada44 100644
--- a/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideButton.vue
+++ b/packages/frontend/editor-ui/src/components/ParameterInputOverrides/FromAiOverrideButton.vue
@@ -20,7 +20,10 @@ const emit = defineEmits<{
 			data-test-id="from-ai-override-button"
 			@click="emit('click')"
 		>
-			<AiStarsIcon size="large" />
+			<span>
+				<!-- The span wrapping the icon centers it due to reliance on legacy behavior -->
+				<AiStarsIcon size="large" />
+			</span>
 		</N8nButton>
 	</N8nTooltip>
 </template>
diff --git a/packages/frontend/editor-ui/src/components/canvas/elements/buttons/CanvasControlButtons.vue b/packages/frontend/editor-ui/src/components/canvas/elements/buttons/CanvasControlButtons.vue
index 6203716e49..2236b392a8 100644
--- a/packages/frontend/editor-ui/src/components/canvas/elements/buttons/CanvasControlButtons.vue
+++ b/packages/frontend/editor-ui/src/components/canvas/elements/buttons/CanvasControlButtons.vue
@@ -1,9 +1,9 @@
 <script setup lang="ts">
-import { Controls } from '@vue-flow/controls';
 import KeyboardShortcutTooltip from '@/components/KeyboardShortcutTooltip.vue';
 import TidyUpIcon from '@/components/TidyUpIcon.vue';
-import { computed } from 'vue';
 import { useI18n } from '@/composables/useI18n';
+import { Controls } from '@vue-flow/controls';
+import { computed } from 'vue';
 
 const props = withDefaults(
 	defineProps<{
@@ -114,9 +114,8 @@ function onTidyUp() {
 
 <style module lang="scss">
 .iconButton {
-	display: flex;
-	align-items: center;
-	justify-content: center;
+	padding-left: 0;
+	padding-right: 0;
 
 	svg {
 		width: 16px;