Fix tests to use related_policy_ids not related_document_ids