Fix small UI Issue