diff --git a/dist/restore-only/index.js b/dist/restore-only/index.js
index 05e0351..effcc41 100644
--- a/dist/restore-only/index.js
+++ b/dist/restore-only/index.js
@@ -3337,11 +3337,11 @@ function restoreCachev2(paths, primaryKey, restoreKeys, options, enableCrossOsAr
             core.info(`Starting download of artifact to: ${paths[0]}`);
             yield (0, download_cache_1.StreamExtract)(signedDownloadURL.blobs[0].signedUrl, paths[0]);
             core.info(`Artifact download completed successfully.`);
+            return keys[0];
         }
         catch (error) {
             throw new Error(`Unable to download and extract cache: ${error.message}`);
         }
-        return undefined;
     });
 }
 /**
diff --git a/dist/restore/index.js b/dist/restore/index.js
index 7c49969..68b33cd 100644
--- a/dist/restore/index.js
+++ b/dist/restore/index.js
@@ -3337,11 +3337,11 @@ function restoreCachev2(paths, primaryKey, restoreKeys, options, enableCrossOsAr
             core.info(`Starting download of artifact to: ${paths[0]}`);
             yield (0, download_cache_1.StreamExtract)(signedDownloadURL.blobs[0].signedUrl, paths[0]);
             core.info(`Artifact download completed successfully.`);
+            return keys[0];
         }
         catch (error) {
             throw new Error(`Unable to download and extract cache: ${error.message}`);
         }
-        return undefined;
     });
 }
 /**
diff --git a/dist/save-only/index.js b/dist/save-only/index.js
index f58584d..bfe88fe 100644
--- a/dist/save-only/index.js
+++ b/dist/save-only/index.js
@@ -3337,11 +3337,11 @@ function restoreCachev2(paths, primaryKey, restoreKeys, options, enableCrossOsAr
             core.info(`Starting download of artifact to: ${paths[0]}`);
             yield (0, download_cache_1.StreamExtract)(signedDownloadURL.blobs[0].signedUrl, paths[0]);
             core.info(`Artifact download completed successfully.`);
+            return keys[0];
         }
         catch (error) {
             throw new Error(`Unable to download and extract cache: ${error.message}`);
         }
-        return undefined;
     });
 }
 /**
diff --git a/dist/save/index.js b/dist/save/index.js
index 1f4f7b7..51a77f7 100644
--- a/dist/save/index.js
+++ b/dist/save/index.js
@@ -3337,11 +3337,11 @@ function restoreCachev2(paths, primaryKey, restoreKeys, options, enableCrossOsAr
             core.info(`Starting download of artifact to: ${paths[0]}`);
             yield (0, download_cache_1.StreamExtract)(signedDownloadURL.blobs[0].signedUrl, paths[0]);
             core.info(`Artifact download completed successfully.`);
+            return keys[0];
         }
         catch (error) {
             throw new Error(`Unable to download and extract cache: ${error.message}`);
         }
-        return undefined;
     });
 }
 /**