Update for migration