minor uml fix