diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
index bf60237..106c511 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
@@ -382,7 +382,7 @@
 				continue;
 			}
 			
-			CallEdgeAttribute callEdgeAttr = new CallEdgeAttribute(callEdge, (ObjectNode) callEdge.getSource(), sourceNodeCell, destinationNodeCell);
+			CallEdgeAttribute callEdgeAttr = new CallEdgeAttribute(callEdge, sourceNode, sourceNodeCell, destinationNodeCell);
 			if (sourceNode instanceof StatefulObjectNode) {
 				graph.insertEdge(layerCell, null, callEdgeAttr, sourceNodeCell, destinationNodeCell, "movable=false;");
 			} else {