Fix migration to remove the original column