diff --git a/src/store/utils/connectedInputs.ts b/src/store/utils/connectedInputs.ts
index b6893424..5676fe30 100644
--- a/src/store/utils/connectedInputs.ts
+++ b/src/store/utils/connectedInputs.ts
@@ -13,6 +13,7 @@ import {
   AnnotationNodeData,
   NanoBananaNodeData,
   GenerateVideoNodeData,
+  Generate3DNodeData,
   VideoStitchNodeData,
   EaseCurveNodeData,
   PromptNodeData,
@@ -66,6 +67,9 @@ function getSourceOutput(sourceNode: WorkflowNode): { type: "image" | "text" | "
       return { type: "3d", value: nbData.output3dUrl };
     }
     return { type: "image", value: nbData.outputImage };
+  } else if (sourceNode.type === "generate3d") {
+    const g3dData = sourceNode.data as Generate3DNodeData;
+    return { type: "3d", value: g3dData.output3dUrl };
   } else if (sourceNode.type === "generateVideo") {
     return { type: "video", value: (sourceNode.data as GenerateVideoNodeData).outputVideo };
   } else if (sourceNode.type === "videoStitch") {