UI menu fix