fix drop column function