Minor UI improvement