A little UI improvement