Fix ui bug