DocumentTrait fixed