Slight UI tweak
Fix for #1010