Fixed small UI bug