protected override void CorrelateMessages(ICorrelationConfig <MySagaData> config) { if (_useMessageContext) { config.CorrelateContext <MyMessage>(context => context.Headers["custom-correlation-id"], d => d.CorrelationId); } else { config.CorrelateHeader <MyMessage>("custom-correlation-id", d => d.CorrelationId); } }