From 9392acbee5a0f5bc24b51ad8dbfbe4a8ec7b9309 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Bosi?=
 <71827178+bosiraphael@users.noreply.github.com>
Date: Tue, 7 Jan 2025 10:21:00 +0100
Subject: [PATCH] Fix use as draft (#9413)

Closes #9408
---
 .../useUseAsDraftWorkflowVersionSingleRecordAction.tsx      | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/packages/twenty-front/src/modules/action-menu/actions/record-actions/single-record/workflow-version-actions/hooks/useUseAsDraftWorkflowVersionSingleRecordAction.tsx b/packages/twenty-front/src/modules/action-menu/actions/record-actions/single-record/workflow-version-actions/hooks/useUseAsDraftWorkflowVersionSingleRecordAction.tsx
index 1ea564255..ead701182 100644
--- a/packages/twenty-front/src/modules/action-menu/actions/record-actions/single-record/workflow-version-actions/hooks/useUseAsDraftWorkflowVersionSingleRecordAction.tsx
+++ b/packages/twenty-front/src/modules/action-menu/actions/record-actions/single-record/workflow-version-actions/hooks/useUseAsDraftWorkflowVersionSingleRecordAction.tsx
@@ -30,19 +30,17 @@ export const useUseAsDraftWorkflowVersionSingleRecordAction: ActionHookWithoutOb
 
     const navigate = useNavigate();
 
-    const workflowStatuses = workflow?.statuses;
+    const hasAlreadyDraftVersion =
+      workflow?.versions.some((version) => version.status === 'DRAFT') || false;
 
     const shouldBeRegistered =
       isDefined(workflowVersion) &&
       isDefined(workflow) &&
-      isDefined(workflowStatuses) &&
       workflowVersion.status !== 'DRAFT';
 
     const onClick = async () => {
       if (!shouldBeRegistered) return;
 
-      const hasAlreadyDraftVersion = workflowStatuses.includes('DRAFT');
-
       if (hasAlreadyDraftVersion) {
         setOpenOverrideWorkflowDraftConfirmationModal(true);
       } else {