fix small UI bug