Location UI fixes