Use correct import for DocumentSet