diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java b/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
index 62c3780..2516a92 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
@@ -181,6 +181,9 @@
                 // Set layout
                 setDAGLayout();
 
+                changeStage(STAGE_DEPENDENCY_MODELING);
+                changeStage(STAGE_PUSH_PULL_SELECTION);
+
                 // Update current file info
                 curFilePath = file.getAbsolutePath();
                 curFileName = file.getName();