🦄 refactor: Revise find()

caoccao · Nov 20, 2024 · eef6998 · eef6998
1 parent 91f5dae
commit eef6998
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/java/com/caoccao/javet/swc4j/ast/interfaces/ISwc4jAst.java b/src/main/java/com/caoccao/javet/swc4j/ast/interfaces/ISwc4jAst.java
@@ -77,13 +77,12 @@ default <T extends ISwc4jAst> List<T> find(Class<T> clazz) {
      * @return the list of AST nodes
      * @since 1.3.0
      */
-    @SuppressWarnings("unchecked")
     default <T extends ISwc4jAst> List<T> find(Class<T> clazz, int depth) {
         AssertionUtils.notNull(clazz, "Class");
         List<T> nodes = SimpleList.of();
         getChildNodes().forEach((childNode) -> {
             if (clazz.isAssignableFrom(childNode.getClass())) {
-                nodes.add((T) childNode);
+                nodes.add(childNode.as(clazz));
             }
             if (depth != 0) {
                 final int newDepth = depth > 0 ? depth - 1 : depth;