diff --git a/src/app/components/workflow-diagram/workflow-diagram.component.html b/src/app/components/workflow-diagram/workflow-diagram.component.html
index 9a67afa..a3d8e4d 100644
--- a/src/app/components/workflow-diagram/workflow-diagram.component.html
+++ b/src/app/components/workflow-diagram/workflow-diagram.component.html
@@ -38,7 +38,7 @@
     </ng-template>
 
     <ng-template #nodeTemplate let-data>
-      @if(data.addInfo.chatWorkflowBlockId !== chatWorkflowBlockTypeEnum.SendTextMessage){
+      @if(data.data.chatWorkflowBlockId !== chatWorkflowBlockTypeEnum.SendTextMessage){
         <div><label style="padding: 5px;">{{ data.data.fieldDetails.label }}</label></div>
       }
       @switch (data.data.chatWorkflowBlockId){