Fixed the migration setup