Minor tweak to UI