Fix rake task error due to column renaming