small fix
new UML