fixing the UMLs