Add cancellation reason back to schema