Minor UI tweak
UI fix