From 3b3255a964cc9af40a399fdc5feee4f3a6f09810 Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Wed, 4 Feb 2015 08:24:38 +1100
Subject: [PATCH] delegate removal of nodes in traversal to exitNode

---
 lib/6to5/traverse/index.js | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/lib/6to5/traverse/index.js b/lib/6to5/traverse/index.js
index d34b8155c8..91cb138f29 100644
--- a/lib/6to5/traverse/index.js
+++ b/lib/6to5/traverse/index.js
@@ -76,11 +76,6 @@ TraversalContext.prototype.enterNode = function (obj, key, node, enter, parent,
     }
   }
 
-  if (this.shouldRemove) {
-    obj[key] = null;
-    this.shouldFlatten = true;
-  }
-
   return node;
 };
 
@@ -97,6 +92,11 @@ TraversalContext.prototype.exitNode = function (obj, key, node, exit, parent, sc
     }
   }
 
+  if (this.shouldRemove) {
+    obj[key] = null;
+    this.shouldFlatten = true;
+  }
+
   return node;
 };