fixed the uml