Handle custom column names in migration