diff --git a/lib/6to5/util.js b/lib/6to5/util.js
index 9f007d06f7..f73fe6db7e 100644
--- a/lib/6to5/util.js
+++ b/lib/6to5/util.js
@@ -147,26 +147,16 @@ exports.template = function (name, nodes, keepExpression) {
     });
   }
 
-  var normaliseNode = function (node) {
-    if (!keepExpression && node.type === "ExpressionStatement") {
-      return node.expression;
-    } else {
-      return node;
-    }
-  };
+  var node = template.body[0];
 
-  var body = template.body;
-
-  if (body.length <= 1) {
-    return normaliseNode(body[0]);
+  if (!keepExpression && node.type === "ExpressionStatement") {
+    return node.expression;
   } else {
-    return body.map(normaliseNode);
+    return node;
   }
 };
 
 exports.codeFrame = function (lines, lineNumber, colNumber) {
-  if (!lineNumber) return "";
-
   colNumber = Math.max(colNumber, 0);
 
   lines = lines.split("\n");