fix a bug in UI;