ASTNode (@NotNull ASTNode element) { if (!is(element, BASIC_IF_STATEMENT)) { return null; } for (ASTNode child : getChildren(element)) { if (is(child, STATEMENT_SET)) { return child; } } return null; }