diff --git a/packages/babel-plugin-transform-runtime/src/index.js b/packages/babel-plugin-transform-runtime/src/index.js
index 6a5e2e4219..b40e67ca3a 100644
--- a/packages/babel-plugin-transform-runtime/src/index.js
+++ b/packages/babel-plugin-transform-runtime/src/index.js
@@ -24,11 +24,9 @@ export default function ({ types: t }) {
 
     visitor: {
       ReferencedIdentifier(path, state) {
-        if (state.opts.regenerator === false) return;
-        
         let { node, parent, scope } = path;
 
-        if (node.name === "regeneratorRuntime") {
+        if (node.name === "regeneratorRuntime" && state.opts.regenerator !== false) {
           path.replaceWith(state.get("regeneratorIdentifier"));
           return;
         }