1344 - Fix branch logic

Author: ivicac

server/libs/atlas/atlas-configuration/atlas-configuration-api/src/main/java/com/bytechef/atlas/configuration/domain/Workflow.java

@@ -419,6 +419,17 @@ private static List<WorkflowTask> getTasks(List<WorkflowTask> workflowTasks, Str
                                 .toList();
 
                             returnedWorkflowTasks.addAll(getTasks(curWorkflowTasks, lastWorkflowNodeName));
+                        } else if (firstItem instanceof Map<?, ?> map &&
+                            map.containsKey(WorkflowConstants.TASKS)) {
+
+                            for (Object curItem : curList) {
+                                Map<String, ?> curMap = (Map<String, ?>) curItem;
+
+                                List<WorkflowTask> curWorkflowTasks = MapUtils.getList(
+                                    curMap, WorkflowConstants.TASKS, WorkflowTask.class, List.of());
+
+                                returnedWorkflowTasks.addAll(getTasks(curWorkflowTasks, lastWorkflowNodeName));
+                            }
                         }
                     }
                 } else if (entry.getValue() instanceof Map<?, ?> curMap) {

server/libs/modules/task-dispatchers/branch/src/main/java/com/bytechef/task/dispatcher/branch/completion/BranchTaskCompletionHandler.java

@@ -131,9 +131,9 @@ public void handle(TaskExecution taskExecution) {
             subTaskExecution = taskExecutionService.create(subTaskExecution);
 
             contextService.push(
-                Validate.notNull(taskExecution.getId(), "id"), Classname.TASK_EXECUTION,
+                Validate.notNull(subTaskExecution.getId(), "id"), Classname.TASK_EXECUTION,
                 taskFileStorage.storeContextValue(
-                    Validate.notNull(taskExecution.getId(), "id"), Classname.TASK_EXECUTION, context));
+                    Validate.notNull(subTaskExecution.getId(), "id"), Classname.TASK_EXECUTION, context));
 
             taskDispatcher.dispatch(subTaskExecution);
         }