Fix compilation error
diff --git a/org.eclipse.epf.diagram.tests/src/org/eclipse/epf/diagram/tests/DiagramUIServiceTest.java b/org.eclipse.epf.diagram.tests/src/org/eclipse/epf/diagram/tests/DiagramUIServiceTest.java
index 04cea33..594ceed 100644
--- a/org.eclipse.epf.diagram.tests/src/org/eclipse/epf/diagram/tests/DiagramUIServiceTest.java
+++ b/org.eclipse.epf.diagram.tests/src/org/eclipse/epf/diagram/tests/DiagramUIServiceTest.java
@@ -82,7 +82,7 @@
 				tasks);
 
 		diagramUIService = new DiagramUIService();
-		DiagramManager mgr = DiagramManager.getInstance(cp1);
+		DiagramManager mgr = DiagramManager.getInstance(cp1, this);
 		Diagram diagram = null;
 		try {
 			Transaction tx =  mgr.getEditingDomain().startTransaction(false,
@@ -151,7 +151,7 @@
 			e.printStackTrace();
 		} finally {
 			if (mgr != null) {
-				mgr.release();
+				mgr.removeConsumer(this);
 			}
 		}		
 	}