diff --git a/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java b/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java
index ce8c7f824..fba0e4154 100644
--- a/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java
+++ b/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java
@@ -384,10 +384,10 @@ public static Map<WorkflowIdentifier, WorkflowTemplate> collectSubWorkflows(
   }
 
   public static Map<TaskIdentifier, TaskTemplate> collectDynamicWorkflowTasks(
-      List<Node> rewrittenNodes,
+      List<Node> nodes,
       Map<TaskIdentifier, TaskTemplate> allTasks,
       Function<TaskIdentifier, TaskTemplate> remoteTaskTemplateFetcher) {
-    return collectTaskIds(rewrittenNodes).stream()
+    return collectTaskIds(nodes).stream()
         // all identifiers should be rewritten at this point
         .map(
             taskId ->
diff --git a/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java b/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java
index 51d24e8db..4af997c22 100644
--- a/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java
+++ b/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java
@@ -309,7 +309,7 @@ private static List<Node> collectTaskTemplates(
 
     Map<TaskIdentifier, TaskTemplate> usedTaskTemplates =
         ProjectClosure.collectDynamicWorkflowTasks(
-            nodes, allTaskTemplates, id -> fetchTaskTemplate(flyteAdminClient, id, cache));
+            rewrittenNodes, allTaskTemplates, id -> fetchTaskTemplate(flyteAdminClient, id, cache));
     allUsedTaskTemplates.putAll(usedTaskTemplates);
 
     return rewrittenNodes;