Modify the way of migration