Updates to latest UMA changes