slight ui tweak