Modify for dntcore API change