slight modification to migration