Small ui tweak

Small ui tweak