Add missing data-ui-component attribute