Don't re-compute the enclosing method (#6876)

typetools · Oct 28, 2024 · 07d8845 · 07d8845
1 parent a70e1e9
commit 07d8845
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/framework/src/main/java/org/checkerframework/common/basetype/BaseTypeVisitor.java b/framework/src/main/java/org/checkerframework/common/basetype/BaseTypeVisitor.java
@@ -2262,9 +2262,8 @@ public Void visitReturn(ReturnTree tree, Void p) {
 
     AnnotatedTypeMirror ret = null;
     if (enclosing.getKind() == Tree.Kind.METHOD) {
-
-      MethodTree enclosingMethod = TreePathUtil.enclosingMethod(getCurrentPath());
-      boolean valid = validateTypeOf(enclosing);
+      MethodTree enclosingMethod = (MethodTree) enclosing;
+      boolean valid = validateTypeOf(enclosingMethod);
       if (valid) {
         ret = atypeFactory.getMethodReturnType(enclosingMethod, tree);
       }