Fixed small UI error