Fix ui

Fixed the new Location UI