Minor changes in migrate module