Add 'update_timestamps' during reassigning and rebalancing