diff --git a/codeflash/languages/java/context.py b/codeflash/languages/java/context.py
index a5597351c..e9ef8054a 100644
--- a/codeflash/languages/java/context.py
+++ b/codeflash/languages/java/context.py
@@ -19,6 +19,8 @@
 if TYPE_CHECKING:
     from tree_sitter import Node
 
+_PARENT_TYPE_NAMES = frozenset(("ClassDef", "InterfaceDef", "EnumDef"))
+
 logger = logging.getLogger(__name__)
 
 
@@ -137,7 +139,7 @@ def _get_parent_type_name(function: FunctionInfo) -> str | None:
     # Check parents for interface/enum
     if function.parents:
         for parent in function.parents:
-            if parent.type in ("ClassDef", "InterfaceDef", "EnumDef"):
+            if parent.type in _PARENT_TYPE_NAMES:
                 return parent.name
 
     return None