diff --git a/src/components/nodes/GenerateImageNode.tsx b/src/components/nodes/GenerateImageNode.tsx
index cdcd14fc..49ca3242 100644
--- a/src/components/nodes/GenerateImageNode.tsx
+++ b/src/components/nodes/GenerateImageNode.tsx
@@ -465,7 +465,9 @@ export function GenerateImageNodeView({ id, data, selected }: GenerateImageNodeV
           onClick: (event) => {
             event.preventDefault();
             event.stopPropagation();
-            downloadMedia(downloadImage, "image").catch(() => {});
+            if (downloadImage) {
+              downloadMedia(downloadImage, "image").catch(() => {});
+            }
           },
         },
         {