Add support for DTS stage