Integrate with Sri's new DTO