Also move this into ShuffleWriterGroup