diff --git a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramCanvasEditable.tsx b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramCanvasEditable.tsx
index 43fccb3b6e..a1c2bbf867 100644
--- a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramCanvasEditable.tsx
+++ b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramCanvasEditable.tsx
@@ -30,6 +30,7 @@ import {
   ReactFlowProvider,
   type Connection,
   type Edge,
+  type OnNodeDrag,
 } from '@xyflow/react';
 import { useCallback } from 'react';
 import { isDefined } from 'twenty-shared/utils';
@@ -133,10 +134,7 @@ export const WorkflowDiagramCanvasEditable = () => {
     });
   };
 
-  const onNodeDragStop = async (
-    _: MouseEvent | TouchEvent,
-    node: WorkflowDiagramNode,
-  ) => {
+  const onNodeDragStop: OnNodeDrag<WorkflowDiagramNode> = async (_, node) => {
     const stepToUpdate =
       workflowWithCurrentVersion?.currentVersion?.steps?.find(
         (step) => step.id === node.id,