Update for new migrations