small ux improvement