minor fixes to location migration change