Fixes the drive migration errors