Modified migration file