Fix small UI bug