[#358] fixes UML