fix a little more UI