minor ux fix