* Fix uml graph
Merge pull request #291 from iseppi/fix-umlgraph

 * Fix uml graph