diff --git a/web/js/image_chooser/chooser.js b/web/js/image_chooser/chooser.js
index 2902c0d..4ef5247 100644
--- a/web/js/image_chooser/chooser.js
+++ b/web/js/image_chooser/chooser.js
@@ -47,7 +47,7 @@ class chooserImageDialog extends ComfyDialog {
     createButtons() {
         const btns = super.createButtons();
         btns[0].onclick = _ => {
-            cancelButtonPressed()
+            if (FlowState.running()) { send_cancel();}
             super.close()
         }
         btns.unshift($el('button', {