use self._section_enum in a few places we missed

rthalley · Dec 24, 2023 · f03f3b7 · f03f3b7
1 parent 63aa46c
commit f03f3b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dns/message.py b/dns/message.py
@@ -393,7 +393,7 @@ def find_rrset(
             section_number = section
             section = self.section_from_number(section_number)
         elif isinstance(section, str):
-            section_number = MessageSection.from_text(section)
+            section_number = self._section_enum.from_text(section)
             section = self.section_from_number(section_number)
         else:
             section_number = self.section_number(section)
@@ -505,7 +505,7 @@ def section_count(self, section: SectionType) -> int:
             section_number = section
             section = self.section_from_number(section_number)
         elif isinstance(section, str):
-            section_number = MessageSection.from_text(section)
+            section_number = self._section_enum.from_text(section)
             section = self.section_from_number(section_number)
         else:
             section_number = self.section_number(section)