refactor: refactor visitCtForEach method (#5244)

INRIA · Jun 11, 2023 · 779438a · 779438a
1 parent bc333f9
commit 779438a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/main/java/spoon/support/compiler/jdt/ParentExiter.java b/src/main/java/spoon/support/compiler/jdt/ParentExiter.java
@@ -654,14 +654,13 @@ private boolean isContainedInForCondition() {
 
 	@Override
 	public void visitCtForEach(CtForEach foreach) {
-		if (foreach.getVariable() == null && child instanceof CtVariable) {
+		if (foreach.getVariable() == null && child instanceof CtLocalVariable<?>) {
 			foreach.setVariable((CtLocalVariable<?>) child);
-			return;
 		} else if (foreach.getExpression() == null && child instanceof CtExpression) {
 			foreach.setExpression((CtExpression<?>) child);
-			return;
+		} else {
+			super.visitCtForEach(foreach);
 		}
-		super.visitCtForEach(foreach);
 	}
 
 	@Override