Replace Context DTO with an actual class