Fix location import