Refactore for new DCG API.