diff --git a/action.yml b/action.yml
index 85685201..0ae276a9 100644
--- a/action.yml
+++ b/action.yml
@@ -33,10 +33,10 @@ inputs:
     description: "Used to specify the version of pip to install with the Python. Supported format: major[.minor][.patch]."
   pip-install:
     description: "Used to specify the packages to install with pip after setting up Python. Can be a requirements file or package names."
-  preclean-pip:
+  preclean:
     description: "When 'true', removes all existing pip packages before installing new ones."
     default: false
-  postclean-pip:
+  postclean:
     description: "When 'true', removes all pip packages installed by this action after the action completes."
     default: false
 outputs:
@@ -49,7 +49,7 @@ outputs:
 runs:
   using: 'node24'
   main: 'dist/setup/index.js'
-  post: 'dist/cache-save/index.js'
+  post: 'dist/post-python/index.js'
   post-if: success()
 branding:
   icon: 'code'
diff --git a/src/post-python.ts b/src/post-python.ts
index 41abd179..02600b66 100644
--- a/src/post-python.ts
+++ b/src/post-python.ts
@@ -19,16 +19,12 @@ export async function run(earlyExit?: boolean) {
       if (earlyExit) {
         process.exit(0);
       }
-      // Optionally clean up pip packages after the post-action if requested.
-      // This mirrors the `preclean-pip` behavior used in the main action.
-      try {
-        const postcleanPip = core.getBooleanInput('postclean-pip');
-        if (postcleanPip) {
-          await cleanPipPackages();
-        }
-      } catch (err) {
-        // getBooleanInput throws if input missing in some contexts; ignore and continue
-      }
+    }
+    // Optionally clean up pip packages after the post-action if requested.
+    // This mirrors the `preclean-pip` behavior used in the main action.
+    const postcleanPip = core.getBooleanInput('postclean');
+    if (postcleanPip) {
+      await cleanPipPackages();
     }
   } catch (error) {
     const err = error as Error;
diff --git a/src/setup-python.ts b/src/setup-python.ts
index f6157340..81d6e5dc 100644
--- a/src/setup-python.ts
+++ b/src/setup-python.ts
@@ -160,7 +160,7 @@ async function run() {
       if (cache && isCacheFeatureAvailable()) {
         await cacheDependencies(cache, pythonVersion);
       }
-      const precleanPip = core.getBooleanInput('preclean-pip');
+      const precleanPip = core.getBooleanInput('preclean');
       if (precleanPip) {
         await cleanPipPackages();
       }