fix ui bug