https://bugs.eclipse.org/bugs/show_bug.cgi?id=151741
diff --git a/plugins/org.eclipse.epf.diagram/src/org/eclipse/epf/diagram/model/impl/NodeImpl.java b/plugins/org.eclipse.epf.diagram/src/org/eclipse/epf/diagram/model/impl/NodeImpl.java
index a08652c..028cfaa 100755
--- a/plugins/org.eclipse.epf.diagram/src/org/eclipse/epf/diagram/model/impl/NodeImpl.java
+++ b/plugins/org.eclipse.epf.diagram/src/org/eclipse/epf/diagram/model/impl/NodeImpl.java
@@ -36,6 +36,7 @@
 import org.eclipse.epf.diagram.model.Link;
 import org.eclipse.epf.diagram.model.ModelPackage;
 import org.eclipse.epf.diagram.model.Node;
+import org.eclipse.epf.diagram.model.WorkProductComposite;
 import org.eclipse.epf.diagram.model.impl.NamedNodeImpl.MethodElementAdapter;
 import org.eclipse.epf.diagram.model.util.GraphicalDataHelper;
 import org.eclipse.epf.diagram.model.util.GraphicalDataManager;
@@ -641,8 +642,12 @@
 
 		this.object = newObject;
 		MethodElement element = (MethodElement) newObject;
-		graphNode = GraphicalDataManager.getInstance().getGraphNode(
+		if(this instanceof WorkProductComposite){
+			graphNode =findGraphNode(getUMADiagram(), element);
+		}else{
+			graphNode = GraphicalDataManager.getInstance().getGraphNode(
 				getUMADiagram(), element);
+		}
 
 		// System.out.println("NodeImpl.setObject(): element: '" +
 		// element.getName() + "', graphNode: " + graphNode);
@@ -677,6 +682,11 @@
 		this.eAdapters().add(createNodeAdapter());
 	}
 
+	protected GraphNode findGraphNode(org.eclipse.epf.uma.Diagram diagram2, Object element) {
+		// TODO Auto-generated method stub
+		return null;
+	}
+
 	/*
 	 * (non-Javadoc)
 	 *