Fix ui bug.