diff --git a/lib/transform.js b/lib/transform.js index 8008e57144..dcde62c2b7 100644 --- a/lib/transform.js +++ b/lib/transform.js @@ -60,12 +60,9 @@ TreeTransformer.prototype = new TreeWalker; tw.push(this); if (tw.before) x = tw.before(this, descend, in_list); if (x === undefined) { - if (!tw.after) { - x = this; - descend(x, tw); - } else { - tw.stack[tw.stack.length - 1] = x = this; - descend(x, tw); + x = this; + descend(x, tw); + if (tw.after) { y = tw.after(x, in_list); if (y !== undefined) x = y; }