bumped to integrate with dataset agent loader