Yes, it's just `entity` to be replaced by `document`