diff --git a/DashAI/front/src/components/generative/SelectTaskMenu.jsx b/DashAI/front/src/components/generative/SelectTaskMenu.jsx
index ef863cd91..9cd825be5 100644
--- a/DashAI/front/src/components/generative/SelectTaskMenu.jsx
+++ b/DashAI/front/src/components/generative/SelectTaskMenu.jsx
@@ -6,7 +6,12 @@ export default function SelectTaskMenu({ tasks, goToNextStep }) {
 
   return (
     <SelectOptionMenu
-      goToNextStep={() => goToNextStep(task.name, task.display_name)}
+      goToNextStep={(taskName) =>
+        goToNextStep(
+          taskName,
+          tasks.find((t) => t.name === taskName).display_name,
+        )
+      }
       title={t("generative:label.generativeModule")}
       subtitle={t("generative:label.selectGenerativeTask")}
       options={tasks.map((task) => ({