diff --git a/lib/main.js b/lib/main.js
index 76d52c0..8a7eb2e 100644
--- a/lib/main.js
+++ b/lib/main.js
@@ -24,8 +24,8 @@ function run() {
             const version = core.getInput('rust-version');
             if (version) {
                 yield rustup.install(version);
-                exec.exec('rustup', ['default', version]);
-                exec.exec('rustup', ['update', version]);
+                yield exec.exec('rustup', ['default', version]);
+                yield exec.exec('rustup', ['update', version]);
             }
         }
         catch (error) {
diff --git a/src/main.ts b/src/main.ts
index 3984b4d..2b1852a 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -9,9 +9,8 @@ async function run() {
 
     if(version) {
       await rustup.install(version);
-
-      exec.exec('rustup', ['default', version]);
-      exec.exec('rustup', ['update', version]);
+      await exec.exec('rustup', ['default', version]);
+      await exec.exec('rustup', ['update', version]);
     }
   } catch (error) {
     core.setFailed(error.message);