Fixed the migration