From 1ff63329a3babd08dd76375204989629f377988d Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Tue, 21 Jul 2015 18:50:08 +0100
Subject: [PATCH] fix directive newline insertion

---
 packages/babel/src/generation/node/whitespace.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/babel/src/generation/node/whitespace.js b/packages/babel/src/generation/node/whitespace.js
index 6de75c7b1d..729dcfbf41 100644
--- a/packages/babel/src/generation/node/whitespace.js
+++ b/packages/babel/src/generation/node/whitespace.js
@@ -104,7 +104,7 @@ exports.nodes = {
    */
 
   Literal(node, parent) {
-    if (t.isExpressionStatement(parent)) {
+    if (t.isExpressionStatement(parent) || t.isProgram(parent)) {
       return {
         after: true
       };