Index: src/org/jboss/tools/smooks/graphical/editors/model/freemarker/FreemarkerTemplateConnection.java =================================================================== --- src/org/jboss/tools/smooks/graphical/editors/model/freemarker/FreemarkerTemplateConnection.java (revision 27401) +++ src/org/jboss/tools/smooks/graphical/editors/model/freemarker/FreemarkerTemplateConnection.java (working copy) @@ -12,6 +12,7 @@ import java.util.List; +import org.jboss.tools.smooks.gef.model.AbstractSmooksGraphicalModel; import org.jboss.tools.smooks.gef.tree.model.TreeNodeConnection; import org.jboss.tools.smooks.templating.template.Mapping; @@ -22,6 +23,21 @@ public class FreemarkerTemplateConnection extends TreeNodeConnection { protected List removeMappings; + + + + public FreemarkerTemplateConnection() { + super(); + // TODO Auto-generated constructor stub + } + + public FreemarkerTemplateConnection( + AbstractSmooksGraphicalModel sourceNode, + AbstractSmooksGraphicalModel targetNode) { + super(sourceNode, targetNode); + // TODO Auto-generated constructor stub + } + public void setRemoveMappingConnections(List removeMappings) { this.removeMappings = removeMappings; } Index: src/org/jboss/tools/smooks/graphical/editors/template/SmooksFreemarkerTemplateGraphicalEditor.java =================================================================== --- src/org/jboss/tools/smooks/graphical/editors/template/SmooksFreemarkerTemplateGraphicalEditor.java (revision 27346) +++ src/org/jboss/tools/smooks/graphical/editors/template/SmooksFreemarkerTemplateGraphicalEditor.java (working copy) @@ -33,6 +33,7 @@ import org.jboss.tools.smooks.graphical.editors.editparts.freemarker.FreemarkerAutoLayout; import org.jboss.tools.smooks.graphical.editors.model.IValidatableModel; import org.jboss.tools.smooks.graphical.editors.model.InputDataContianerModel; +import org.jboss.tools.smooks.graphical.editors.model.freemarker.FreemarkerTemplateConnection; import org.jboss.tools.smooks.graphical.editors.model.freemarker.FreemarkerTemplateGraphicalModel; import org.jboss.tools.smooks.graphical.editors.model.freemarker.IFreemarkerTemplateModel; import org.jboss.tools.smooks.graphical.editors.process.TaskType; @@ -113,7 +114,7 @@ TreeNodeModel connectionTarget = freemarkerModel.getModelNode(mapping.getMappingNode()); if(connectionSource != null && connectionTarget != null) { - TreeNodeConnection connection = new TreeNodeConnection(connectionSource, connectionTarget); + TreeNodeConnection connection = new FreemarkerTemplateConnection(connectionSource, connectionTarget); connectionSource.getSourceConnections().add(connection); connectionSource.fireConnectionChanged();