diff --git a/src/components/graph/selectionToolbox/MaskEditorButton.vue b/src/components/graph/selectionToolbox/MaskEditorButton.vue
index 73ec9cc67b..077a2acd7b 100644
--- a/src/components/graph/selectionToolbox/MaskEditorButton.vue
+++ b/src/components/graph/selectionToolbox/MaskEditorButton.vue
@@ -7,10 +7,10 @@
     }"
     severity="secondary"
     text
+    icon="pi pi-pencil"
+    icon-class="w-4 h-4"
     @click="openMaskEditor"
-  >
-    <i-comfy:mask class="!h-4 !w-4" />
-  </Button>
+  />
 </template>
 
 <script setup lang="ts">