Minor ui tweak