Small change to UML