diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/PushPullSelectionStage.java b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/PushPullSelectionStage.java
index 31def3f..9ba5b43 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/PushPullSelectionStage.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/PushPullSelectionStage.java
@@ -239,17 +239,17 @@
                     // output edge
                     DataTransferChannel channel = ((ChannelNode) dataFlowEdge.getSource()).getChannel();
                     ResourcePath dstRes = ((ResourceNode) dataFlowEdge.getDestination()).getInSideResource(channel);
-                    graph.insertEdge(parent, null, new Editor.SrcDstAttribute(channel, dstRes), channelOutToCell.get(channel), resNodeToCell.get((ResourceNode) dataFlowEdge.getDestination()), "movable=false");
+                    graph.insertEdge(layer, null, new Editor.SrcDstAttribute(channel, dstRes), channelOutToCell.get(channel), resNodeToCell.get((ResourceNode) dataFlowEdge.getDestination()), "movable=false");
                 } else {
                     // input edge
                     DataTransferChannel channel = ((ChannelNode) dataFlowEdge.getDestination()).getChannel();
                     ResourcePath srcRes = ((ResourceNode) dataFlowEdge.getSource()).getOutSideResource(channel);
                     Set<Map.Entry<ResourceNode, ResourceNode>> toRes = getResourceDependencyForChannel(channel, dataFlowGraph);
                     for (Map.Entry<ResourceNode, ResourceNode> RtoR : toRes) {
-                        graph.insertEdge(parent, null, null, resNodeToCell.get(RtoR.getValue()), resNodeToCell.get(RtoR.getKey()), "dashed=true;movable=false");
+                        graph.insertEdge(layer, null, null, resNodeToCell.get(RtoR.getValue()), resNodeToCell.get(RtoR.getKey()), "dashed=true;movable=false");
                     }
 
-                    graph.insertEdge(parent, null, new Editor.SrcDstAttribute(srcRes, channel), resNodeToCell.get((ResourceNode) dataFlowEdge.getSource()), channelInToCell.get(channel), "movable=false");
+                    graph.insertEdge(layer, null, new Editor.SrcDstAttribute(srcRes, channel), resNodeToCell.get((ResourceNode) dataFlowEdge.getSource()), channelInToCell.get(channel), "movable=false");
                 }
             }
 
@@ -257,7 +257,7 @@
                 // reference edges
                 DataTransferChannel channel = (DataTransferChannel) ch;
                 for (ResourcePath refRes : channel.getReferenceResources()) {
-                    graph.insertEdge(parent, null, null, resNodeToCell.get(dataFlowGraph.getResourceNode(refRes)), channelInToCell.get(channel), "dashed=true;movable=false");
+                    graph.insertEdge(layer, null, null, resNodeToCell.get(dataFlowGraph.getResourceNode(refRes)), channelInToCell.get(channel), "dashed=true;movable=false");
                 }
             }
         } finally {