Update changes to migration