origStage.getLogDir(), origStage.getClusterName(),
origStage.getRequestContext(), origStage.getClusterHostInfo());
newStage.setSuccessFactors(origStage.getSuccessFactors());
for (RoleGraphNode rgn : stageGraphNodes) {
for (String host : rgn.getHosts()) {
newStage.addExecutionCommandWrapper(origStage, host, rgn.getRole());
}
}
return newStage;
}