making some changes in driver for live migration