Fix remaining uses of 'ukmo__um_version'.
Fix test functions logic.