WIP: change migration to new model.