small UI cleanup