1 files changed, 17 insertions, 1 deletions
diff --git a/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/linklf/router/provider/CustomRoutersDiagramRootEditPart.java b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/linklf/router/provider/CustomRoutersDiagramRootEditPart.java
index fc8dfbcc46f..48ccfdd04db 100644
--- a/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/linklf/router/provider/CustomRoutersDiagramRootEditPart.java
+++ b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/linklf/router/provider/CustomRoutersDiagramRootEditPart.java
@@ -10,6 +10,8 @@ import org.eclipse.gmf.runtime.notation.MeasurementUnit;
 
 public class CustomRoutersDiagramRootEditPart extends RenderedDiagramRootEditPart {
 
+	private CustomRoutersConnectionLayer myConnectionLayer;
+
 	public CustomRoutersDiagramRootEditPart(MeasurementUnit mUnit) {
 		super(mUnit);
 	}
@@ -30,7 +32,21 @@ public class CustomRoutersDiagramRootEditPart extends RenderedDiagramRootEditPar
 	}
 
 	protected ConnectionLayerEx createConnectionLayer() {
-		return new CustomRoutersConnectionLayer();
+		myConnectionLayer = new CustomRoutersConnectionLayer();
+		return myConnectionLayer;
+	}
+
+	@Override
+	protected void register() {
+		super.register();
+		myConnectionLayer.setEditPartViewer(getViewer());
+
+	}
+
+	@Override
+	protected void unregister() {
+		myConnectionLayer.setEditPartViewer(null);
+		super.unregister();
 	}
 
 }