fix merge error

soot-oss · Jan 14, 2025 · 86bf020 · 86bf020
1 parent 4959c49
commit 86bf020
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sootup.core/src/main/java/sootup/core/graph/MutableBlockStmtGraph.java b/sootup.core/src/main/java/sootup/core/graph/MutableBlockStmtGraph.java
@@ -24,6 +24,7 @@
 
 import com.google.common.collect.Lists;
 import java.util.*;
+import java.util.stream.Collectors;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 import org.apache.commons.lang3.tuple.MutablePair;
@@ -1516,6 +1517,12 @@ public BasicBlock<?> getStartingStmtBlock() {
     return getBlockOf(startingStmt);
   }
 
+  @Override
+  @Nonnull
+  public List<BasicBlock<?>> getTailStmtBlocks() {
+    return getTails().stream().map(stmt -> getBlockOf(stmt)).collect(Collectors.toList());
+  }
+
   @Override
   @Nullable
   public BasicBlock<?> getBlockOf(@Nonnull Stmt stmt) {