diff --git a/dist/index.js b/dist/index.js
index 76369e6..933e5a8 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -965,37 +965,42 @@ const common = __webpack_require__(239)
 
 async function run() {
   try {
-    process.chdir(core.getInput('working-directory'))
-    const platform = common.getVirtualEnvironmentName()
-    const [engine, version] = parseRubyEngineAndVersion(core.getInput('ruby-version'))
-
-    let installer
-    if (platform === 'windows-latest' && engine !== 'jruby') {
-      installer = __webpack_require__(826)
-    } else {
-      installer = __webpack_require__(489)
-    }
-
-    const engineVersions = installer.getAvailableVersions(platform, engine)
-    const ruby = validateRubyEngineAndVersion(platform, engineVersions, engine, version)
-
-    createGemRC()
-
-    const [rubyPrefix, newPathEntries] = await installer.install(platform, ruby)
-
-    setupPath(ruby, newPathEntries)
-
-    if (core.getInput('bundler') !== 'none') {
-      await common.measure('Installing Bundler', async () =>
-        installBundler(platform, rubyPrefix, engine, version))
-    }
-
-    core.setOutput('ruby-prefix', rubyPrefix)
+    await main()
   } catch (error) {
     core.setFailed(error.message)
   }
 }
 
+async function main() {
+  process.chdir(core.getInput('working-directory'))
+
+  const platform = common.getVirtualEnvironmentName()
+  const [engine, version] = parseRubyEngineAndVersion(core.getInput('ruby-version'))
+
+  let installer
+  if (platform === 'windows-latest' && engine !== 'jruby') {
+    installer = __webpack_require__(826)
+  } else {
+    installer = __webpack_require__(489)
+  }
+
+  const engineVersions = installer.getAvailableVersions(platform, engine)
+  const ruby = validateRubyEngineAndVersion(platform, engineVersions, engine, version)
+
+  createGemRC()
+
+  const [rubyPrefix, newPathEntries] = await installer.install(platform, ruby)
+
+  setupPath(ruby, newPathEntries)
+
+  if (core.getInput('bundler') !== 'none') {
+    await common.measure('Installing Bundler', async () =>
+      installBundler(platform, rubyPrefix, engine, version))
+  }
+
+  core.setOutput('ruby-prefix', rubyPrefix)
+}
+
 function parseRubyEngineAndVersion(rubyVersion) {
   if (rubyVersion === 'default') {
     if (fs.existsSync('.ruby-version')) {
diff --git a/index.js b/index.js
index 61d941d..acf1841 100644
--- a/index.js
+++ b/index.js
@@ -7,37 +7,42 @@ const common = require('./common')
 
 export async function run() {
   try {
-    process.chdir(core.getInput('working-directory'))
-    const platform = common.getVirtualEnvironmentName()
-    const [engine, version] = parseRubyEngineAndVersion(core.getInput('ruby-version'))
-
-    let installer
-    if (platform === 'windows-latest' && engine !== 'jruby') {
-      installer = require('./windows')
-    } else {
-      installer = require('./ruby-builder')
-    }
-
-    const engineVersions = installer.getAvailableVersions(platform, engine)
-    const ruby = validateRubyEngineAndVersion(platform, engineVersions, engine, version)
-
-    createGemRC()
-
-    const [rubyPrefix, newPathEntries] = await installer.install(platform, ruby)
-
-    setupPath(ruby, newPathEntries)
-
-    if (core.getInput('bundler') !== 'none') {
-      await common.measure('Installing Bundler', async () =>
-        installBundler(platform, rubyPrefix, engine, version))
-    }
-
-    core.setOutput('ruby-prefix', rubyPrefix)
+    await main()
   } catch (error) {
     core.setFailed(error.message)
   }
 }
 
+async function main() {
+  process.chdir(core.getInput('working-directory'))
+
+  const platform = common.getVirtualEnvironmentName()
+  const [engine, version] = parseRubyEngineAndVersion(core.getInput('ruby-version'))
+
+  let installer
+  if (platform === 'windows-latest' && engine !== 'jruby') {
+    installer = require('./windows')
+  } else {
+    installer = require('./ruby-builder')
+  }
+
+  const engineVersions = installer.getAvailableVersions(platform, engine)
+  const ruby = validateRubyEngineAndVersion(platform, engineVersions, engine, version)
+
+  createGemRC()
+
+  const [rubyPrefix, newPathEntries] = await installer.install(platform, ruby)
+
+  setupPath(ruby, newPathEntries)
+
+  if (core.getInput('bundler') !== 'none') {
+    await common.measure('Installing Bundler', async () =>
+      installBundler(platform, rubyPrefix, engine, version))
+  }
+
+  core.setOutput('ruby-prefix', rubyPrefix)
+}
+
 function parseRubyEngineAndVersion(rubyVersion) {
   if (rubyVersion === 'default') {
     if (fs.existsSync('.ruby-version')) {