Fix live migration method