A small ui tweak