small UI improvement