diff --git a/packages/babel/src/helpers/code-frame.js b/packages/babel/src/helpers/code-frame.js
index fbed6262bf..5028248c57 100644
--- a/packages/babel/src/helpers/code-frame.js
+++ b/packages/babel/src/helpers/code-frame.js
@@ -108,8 +108,8 @@ export default function (lines: number, lineNumber: number, colNumber: number, o
   }).join("\n");
 
   if (highlighted) {
-    frame = chalk.reset() + frame;
+    return chalk.reset(frame);
+  } else {
+    return frame;
   }
-
-  return frame;
 }