Fixed callback with update_columns method