fixed a minor bug in UI