diff --git a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramEffect.tsx b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramEffect.tsx
index 2ce093d09..bfee6650c 100644
--- a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramEffect.tsx
+++ b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramEffect.tsx
@@ -47,13 +47,10 @@ export const WorkflowDiagramEffect = ({
         if (isDefined(lastCreatedStepId)) {
           mergedWorkflowDiagram.nodes = mergedWorkflowDiagram.nodes.map(
             (node) => {
-              if (node.id === lastCreatedStepId) {
-                return {
-                  ...node,
-                  selected: true,
-                };
-              }
-              return node;
+              return {
+                ...node,
+                selected: node.id === lastCreatedStepId,
+              };
             },
           );