diff --git a/src/designer/designer_packages/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx b/src/designer/designer_packages/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
index 713ae832c..35eccbe1c 100644
--- a/src/designer/designer_packages/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
+++ b/src/designer/designer_packages/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
@@ -143,7 +143,7 @@ export class FlowchartComponent {
       edges.push({id: index, source: connection.source, target: connection.target});
     });
 
-    let data = {nodes: nodes, edges: edges}
+    let data = {nodes: nodes, edges: edges.map(e => ({...e, source: {...e.source, cell: e.source.activity}, target: {...e.target, cell: e.target.activity}}))}
     let newLayout = dagreLayout.layout(data);
 
     newLayout.nodes.forEach(node => {
diff --git a/src/modules/Elsa.Workflows.Designer/package.json b/src/modules/Elsa.Workflows.Designer/package.json
index 4da3726bc..aa2982701 100644
--- a/src/modules/Elsa.Workflows.Designer/package.json
+++ b/src/modules/Elsa.Workflows.Designer/package.json
@@ -6,7 +6,7 @@
     "build:debug": "cross-env webpack --mode=development --env configuration=Debug"
   },
   "devDependencies": {
-    "@elsa-workflows/elsa-workflows-designer": "3.0.3-preview.208",
+    "@elsa-workflows/elsa-workflows-designer": "3.0.3-preview.209",
     "copy-webpack-plugin": "^11.0.0",
     "cross-env": "^7.0.3",
     "monaco-editor": "^0.36.1",