diff --git a/components/canvas/nodes/ImageNode.tsx b/components/canvas/nodes/ImageNode.tsx
index 7242e3b57..abf78862d 100644
--- a/components/canvas/nodes/ImageNode.tsx
+++ b/components/canvas/nodes/ImageNode.tsx
@@ -378,7 +378,6 @@ export function ImageNode({ node, isSelected }: ImageNodeProps) {
       setTextEditPanelOpen(false);
       setIsEditingTitle(false);
       setPromptDialogOpen(false);
-      setIsCompareMode(false);
     }
   }, [isSelected]);
 
@@ -1713,11 +1712,11 @@ export function ImageNode({ node, isSelected }: ImageNodeProps) {
               </div>
             )}
 
-            {isSelected && !isMultiSelected && hasCompareSource && !isThisNodePicking && (
+            {hasCompareSource && !isThisNodePicking && (isCompareMode || (isSelected && !isMultiSelected)) && (
               <button
                 type="button"
                 className={cn(
-                  "absolute right-2 top-1/2 z-[220] inline-flex h-14 w-14 -translate-y-1/2 items-center justify-center rounded-full border text-white transition-all data-[state=on]:scale-100",
+                  "pointer-events-auto absolute right-2 top-1/2 z-[220] inline-flex h-14 w-14 -translate-y-1/2 items-center justify-center rounded-full border text-white transition-all data-[state=on]:scale-100",
                   isCompareMode
                     ? "border-sky-100 bg-sky-500/95 shadow-[0_0_0_3px_rgba(255,255,255,0.55),0_10px_22px_rgba(14,165,233,0.42)]"
                     : "border-white/80 bg-black/78 shadow-[0_8px_18px_rgba(0,0,0,0.34)] hover:bg-black/88"