Fix build after integration of UML discovery