Mods to make UI rendering work