Minor fix in setting shard hint in context