correct ui bug