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
       };