https://bugs.eclipse.org/bugs/show_bug.cgi?id=151493
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 028cfaa..5f0d22d 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
@@ -642,8 +642,12 @@
 
 		this.object = newObject;
 		MethodElement element = (MethodElement) newObject;
+		
+		// This workproduct composite is necessary, because
+		// WorkproductComposite object is task, and there two types of composite.
+		// have to find exact workproductcomposite graphnode. 
 		if(this instanceof WorkProductComposite){
-			graphNode =findGraphNode(getUMADiagram(), element);
+			graphNode =getGraphNode(element);
 		}else{
 			graphNode = GraphicalDataManager.getInstance().getGraphNode(
 				getUMADiagram(), element);
@@ -682,7 +686,7 @@
 		this.eAdapters().add(createNodeAdapter());
 	}
 
-	protected GraphNode findGraphNode(org.eclipse.epf.uma.Diagram diagram2, Object element) {
+	public GraphNode getGraphNode(MethodElement methodElement){
 		// TODO Auto-generated method stub
 		return null;
 	}