A minor bugfix for the uml extension