Fix bug in ui_schema