diff --git a/src/ds_caselaw_utils/courts.py b/src/ds_caselaw_utils/courts.py index e4f6e08..13cce76 100644 --- a/src/ds_caselaw_utils/courts.py +++ b/src/ds_caselaw_utils/courts.py @@ -66,8 +66,8 @@ def __init__(self, court: Court, jurisdiction: Jurisdiction) -> None: self.jurisdictions: list[Jurisdiction] = [] @property - def code(self) -> str: - return "/".join((self.court.code, self.jurisdiction.code)) + def code(self) -> CourtCode: + return CourtCode("/".join((self.court.code, self.jurisdiction.code))) @property def name(self) -> str: