diff --git a/web/extensions/core/maskeditor.js b/web/extensions/core/maskeditor.js
index 1d28de490..2f61a2121 100644
--- a/web/extensions/core/maskeditor.js
+++ b/web/extensions/core/maskeditor.js
@@ -187,7 +187,13 @@ class MaskEditorDialog extends ComfyDialog {
 			document.removeEventListener("keydown", MaskEditorDialog.handleKeyDown);
 			self.close();
 		});
-		var saveButton = this.createRightButton("Save", () => {
+
+        var save_button_title = "Save";
+		if(ComfyApp.clipspace_return_node) {
+            save_button_title = "Save to node";
+		}
+
+		var saveButton = this.createRightButton(save_button_title, () => {
 			document.removeEventListener("mouseup", MaskEditorDialog.handleMouseUp);
 			document.removeEventListener("keydown", MaskEditorDialog.handleKeyDown);
 				self.save();
diff --git a/web/scripts/app.js b/web/scripts/app.js
index 9f659e8eb..6cf724bc2 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -59,6 +59,7 @@ export class ComfyApp {
     static onClipspaceEditorSaved() {
         if(ComfyApp.clipspace_return_node) {
             ComfyApp.pasteToClipspace(ComfyApp.clipspace_return_node);
+            ComfyApp.clipspace_return_node = null;
         }
     }