diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/DependencyModelingStage.java b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/DependencyModelingStage.java
index 406c148..7c597ef 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/DependencyModelingStage.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/DependencyModelingStage.java
@@ -14,6 +14,7 @@
 import generators.JavaCodeGenerator;
 import models.Edge;
 import models.algebra.*;
+import models.controlFlowModel.ControlFlowGraph;
 import models.dataConstraintModel.Channel;
 import models.dataConstraintModel.ChannelMember;
 import models.dataConstraintModel.ResourcePath;
@@ -73,7 +74,8 @@
             model = prevStage.getModel();
 
             DataFlowGraph dataFlowGraph = ((PushPullSelectionStage) prevStage).getDataFlowGraph();
-            dependencyGraph = new DependencyGraph(dataFlowGraph, model);
+            ControlFlowGraph controlFlowGraph = new ControlFlowGraph(dataFlowGraph, model);
+            dependencyGraph = new DependencyGraph(controlFlowGraph, model);
 
             resNodeToCell = ((PushPullSelectionStage) prevStage).getResNodeToCell();
             eventChannelToCell = ((PushPullSelectionStage) prevStage).getEventChannelToCell();