diff --git a/dist/save/index.js b/dist/save/index.js
index bd9f422..e8ff428 100644
--- a/dist/save/index.js
+++ b/dist/save/index.js
@@ -47298,7 +47298,8 @@ 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(constants_1.State.CachePrimaryKey);
+            //const primaryKey = core.getState(State.CachePrimaryKey);
+            const primaryKey = core.getInput(constants_1.Inputs.Key);
             if (!primaryKey) {
                 utils.logWarning(`Error retrieving key from state.`);
                 return;
diff --git a/src/save.ts b/src/save.ts
index a0a21bf..7cb4aac 100644
--- a/src/save.ts
+++ b/src/save.ts
@@ -27,7 +27,8 @@ async function run(): Promise<void> {
         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.getState(State.CachePrimaryKey);
+        const primaryKey = core.getInput(Inputs.Key);
         if (!primaryKey) {
             utils.logWarning(`Error retrieving key from state.`);
             return;