From 286fe1ac83638c047ae2793926e264a8e35b687d Mon Sep 17 00:00:00 2001
From: Sankalp Kotewar <98868223+kotewar@users.noreply.github.com>
Date: Tue, 15 Nov 2022 04:41:27 +0000
Subject: [PATCH] Updated dist for save

---
 dist/save/index.js | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/dist/save/index.js b/dist/save/index.js
index e8ff428..6bf1854 100644
--- a/dist/save/index.js
+++ b/dist/save/index.js
@@ -47298,8 +47298,7 @@ function run() {
             }
             const state = utils.getCacheState();
             // Inputs are re-evaluted before the post action, so we want the original key used for restore
-            //const primaryKey = core.getState(State.CachePrimaryKey);
-            const primaryKey = core.getInput(constants_1.Inputs.Key);
+            const primaryKey = core.getState(constants_1.State.CachePrimaryKey) || core.getInput(constants_1.Inputs.Key);
             if (!primaryKey) {
                 utils.logWarning(`Error retrieving key from state.`);
                 return;