diff --git a/src/components/image-editor/useImageCanvasGenerationWorkflow.ts b/src/components/image-editor/useImageCanvasGenerationWorkflow.ts
index 544d32be6..52571154e 100644
--- a/src/components/image-editor/useImageCanvasGenerationWorkflow.ts
+++ b/src/components/image-editor/useImageCanvasGenerationWorkflow.ts
@@ -1746,37 +1746,30 @@ export function useImageCanvasGenerationWorkflow({
     viewport,
   ]);
 
-  const openModel3dTextToModelDialog = useCallback(() => {
-    openPlacedCanvasGenerationDialog(
-      createModel3dGenerationDialogDraft({
-        canvasSize,
-        viewport,
-        mode: 'model3d-text-to-model',
-      }),
-    );
-    activateCanvasGenerationEntry('model3d');
-  }, [
-    activateCanvasGenerationEntry,
-    canvasSize,
-    openPlacedCanvasGenerationDialog,
-    viewport,
-  ]);
+  const openModel3dGenerationDialog = useCallback(
+    (mode: 'model3d-text-to-model' | 'model3d-image-to-model') => {
+      openPlacedCanvasGenerationDialog(
+        createModel3dGenerationDialogDraft({ canvasSize, viewport, mode }),
+      );
+      activateCanvasGenerationEntry('model3d');
+    },
+    [
+      activateCanvasGenerationEntry,
+      canvasSize,
+      openPlacedCanvasGenerationDialog,
+      viewport,
+    ],
+  );
 
-  const openModel3dImageToModelDialog = useCallback(() => {
-    openPlacedCanvasGenerationDialog(
-      createModel3dGenerationDialogDraft({
-        canvasSize,
-        viewport,
-        mode: 'model3d-image-to-model',
-      }),
-    );
-    activateCanvasGenerationEntry('model3d');
-  }, [
-    activateCanvasGenerationEntry,
-    canvasSize,
-    openPlacedCanvasGenerationDialog,
-    viewport,
-  ]);
+  const openModel3dTextToModelDialog = useCallback(
+    () => openModel3dGenerationDialog('model3d-text-to-model'),
+    [openModel3dGenerationDialog],
+  );
+
+  const openModel3dImageToModelDialog = useCallback(
+    () => openModel3dGenerationDialog('model3d-image-to-model'),
+    [openModel3dGenerationDialog],
+  );
 
   const openEditDialog = useCallback(
     (sourceLayer: CanvasLayer) => {