fixed little UI bug