DTO fixes and UT