small ui bug fixed