fix a ui bug