Update column name and tests