From 4f03b31551d9688771db376ff2df5fabd28b7542 Mon Sep 17 00:00:00 2001
From: Ron Buckton <rbuckton@microsoft.com>
Date: Tue, 23 Feb 2016 15:40:12 -0800
Subject: [PATCH] Updated lift behavior

---
 src/compiler/visitor.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/compiler/visitor.ts b/src/compiler/visitor.ts
index 92b00b1270..ead9119e21 100644
--- a/src/compiler/visitor.ts
+++ b/src/compiler/visitor.ts
@@ -502,8 +502,8 @@ namespace ts {
             return node;
         }
 
-        if (lift !== undefined && visited !== undefined && isNodeArrayNode(visited)) {
-            visited = lift((<NodeArrayNode<Node>>visited).nodes);
+        if (visited !== undefined && isNodeArrayNode(visited)) {
+            visited = (lift || extractSingleNode)((<NodeArrayNode<Node>>visited).nodes);
         }
 
         if (parenthesize !== undefined && visited !== undefined) {