diff --git a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll index 0bb2cf66940..a821ea67363 100644 --- a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll +++ b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll @@ -185,8 +185,7 @@ module NeighborhoodBodies { */ Raw::AstNode getNeighborhoodAstNode(Raw::AstNode node) { if - // `node` will always have a parent as we start at and endpoint - node.getParentNode() = getOutermostEnclosingFunction(node) or + node = getOutermostEnclosingFunction(node) or getNumDescendents(node.getParentNode()) > maxNumDescendants() then result = node else result = getNeighborhoodAstNode(node.getParentNode())