small ux fix