diff --git a/dist/restore/index.js b/dist/restore/index.js
index 7ef0ef5..590681e 100644
--- a/dist/restore/index.js
+++ b/dist/restore/index.js
@@ -48991,7 +48991,7 @@ function run() {
             });
             const cacheKey = yield cache.restoreCache(cachePaths, primaryKey, restoreKeys);
             //Check if user wants to save cache despite of failure in any previous job
-            const saveCache = process.env[constants_1.Inputs.SaveCacheOnAnyFailure];
+            const saveCache = core.getInput(constants_1.Inputs.SaveCacheOnAnyFailure);
             if (saveCache === "yes") {
                 core.debug(`save cache input variable is set to yes`);
                 core.exportVariable(constants_1.Variables.SaveCacheOnAnyFailure, saveCache);
diff --git a/src/restore.ts b/src/restore.ts
index 29fb5e1..fe71674 100644
--- a/src/restore.ts
+++ b/src/restore.ts
@@ -36,7 +36,7 @@ async function run(): Promise<void> {
         );
 
         //Check if user wants to save cache despite of failure in any previous job
-        const saveCache = process.env[Inputs.SaveCacheOnAnyFailure];
+        const saveCache = core.getInput(Inputs.SaveCacheOnAnyFailure);
         if (saveCache === "yes") {
             core.debug(`save cache input variable is set to yes`);
             core.exportVariable(Variables.SaveCacheOnAnyFailure, saveCache);