small ux code change