diff --git a/deidentification/deidentification.py b/deidentification/deidentification.py index d67dd19..e2b7082 100644 --- a/deidentification/deidentification.py +++ b/deidentification/deidentification.py @@ -171,7 +171,7 @@ def deidentify_with_wrapped_html(self, text: str, html_begin: str = HTML_BEGIN, return buffer.getvalue() def get_identified_elements(self) -> dict: - elements = {"message": self.replaced_text, "entities": self.aggregate_persons, "pronouns": self.aggregate_pronouns} + elements = {"message": self.text, "entities": self.aggregate_persons, "pronouns": self.aggregate_pronouns} return elements def _find_all_persons(self) -> int: