fixing uml model