ui bug fixed