Update for aggregate remove columns

Update for aggregate remove columns