diff --git a/cleanup.js b/cleanup.js
index 6b7ab7d..a0ffe37 100644
--- a/cleanup.js
+++ b/cleanup.js
@@ -1,8 +1,10 @@
 const core = require('@actions/core');
 const { execFileSync } = require('child_process');
-const { sshAgentCmd } = require('./paths.js');
+const { sshAgentCmdDefault } = require('./paths.js');
 
 try {
+    const sshAgentCmdInput = core.getInput('ssh-agent-cmd');
+    const sshAgentCmd = sshAgentCmdInput ? sshAgentCmdInput : sshAgentCmdDefault;
     // Kill the started SSH agent
     console.log('Stopping SSH agent');
     execFileSync(sshAgentCmd, ['-k'], { stdio: 'inherit' });