diff --git a/AlgebraicDataflowArchitectureModel/src/models/dependencyModel/DependencyGraph.java b/AlgebraicDataflowArchitectureModel/src/models/dependencyModel/DependencyGraph.java
index c8b7145..686c0b3 100644
--- a/AlgebraicDataflowArchitectureModel/src/models/dependencyModel/DependencyGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/models/dependencyModel/DependencyGraph.java
@@ -53,8 +53,8 @@
     public void addEdge(Node srcNode, Node dstNode){
         addNode((ObjectNode) srcNode);
         addNode((ObjectNode) dstNode);
-        addEdge(new DependencyEdge(getOrCreateObjectNode((
-                EventChannelObjectNode) srcNode), getOrCreateObjectNode((StatefulObjectNode)dstNode)));
+        addEdge(new DependencyEdge(getOrCreateObjectNode((ObjectNode)
+                srcNode), getOrCreateObjectNode((ObjectNode) dstNode)));
     }
 
     private ObjectNode getOrCreateObjectNode(ObjectNode node) {