Add shard_key to scope used by `update_columns`