Fixing the migration