Tiny fix
Merged with ui upstream