Modify scaffold migration for Edition