Update to match the current xui framework