diff --git a/src/org/urmiauniversity/it/mst/cpm/CPM.java b/src/org/urmiauniversity/it/mst/cpm/CPM.java
index 301b690..9a920f9 100644
--- a/src/org/urmiauniversity/it/mst/cpm/CPM.java
+++ b/src/org/urmiauniversity/it/mst/cpm/CPM.java
@@ -204,7 +204,11 @@ public void execute(GraphModel gm, AttributeModel am) {
         for (Node vi : nodes) {
             for (Node vj : nodes) {
                 if ((vi != vj) && (getSharedNodes(vi, vj) == k - 1)) {
-                    edges.add(gm.factory().newEdge(vi, vj, 1.0f, false));
+                    if (g.getGraphModel().isDirected()) {
+                        edges.add(gm.factory().newEdge(vi, vj, 1.0f, true));
+                    } else {
+                        edges.add(gm.factory().newEdge(vi, vj, 1.0f, false));
+                    }
                 }
             }
         }