Migration script modified