Added migration for staged entities