diff --git a/src/designer/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx b/src/designer/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
index 84463bcb5..e7f3c317b 100644
--- a/src/designer/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
+++ b/src/designer/elsa-workflows-designer/src/modules/flowchart/flowchart.tsx
@@ -239,8 +239,8 @@ export class FlowchartComponent {
     return this.exportInternal();
   }
 
-  @Watch('workflowDefinition')
-  private onWorkflowDefinitionChanged(value: WorkflowDefinition) {
+  @Watch('rootActivity')
+  private onActivityChanged(value: Activity) {
     this.updateLookups();
   }