Small fix to UML