diff --git a/org.framed.iorm.ui/src/org/framed/iorm/ui/editPolicy/EditPolicyService.java b/org.framed.iorm.ui/src/org/framed/iorm/ui/editPolicy/EditPolicyService.java
index f7cf055172ad89bdb08c1f46d8b51f02e4e21a9a..8c8053e5216d37d4d38933efdf9b492f74326484 100644
--- a/org.framed.iorm.ui/src/org/framed/iorm/ui/editPolicy/EditPolicyService.java
+++ b/org.framed.iorm.ui/src/org/framed/iorm/ui/editPolicy/EditPolicyService.java
@@ -69,7 +69,7 @@ public class EditPolicyService {
 	{ 
 		//System.out.println("updating config for " + diagram.getName());
 		diagram = UIUtil.getMainDiagramForAnyDiagram(diagram);
-		editPolicyHandlers.get(diagram.getName()).updateConfig(editpolicymodels, config);
+		EditPolicyService.getHandler(diagram).updateConfig(editpolicymodels, config);
 	}
 
 	/*