diff --git a/src/helpers.js b/src/helpers.js
index 2c1b2fa..154391e 100644
--- a/src/helpers.js
+++ b/src/helpers.js
@@ -2,6 +2,10 @@ const { existsSync, mkdirSync, writeFileSync } = require('fs');
 const { join } = require('path');
 
 const validateDir = (dir) => {
+  if (!dir) {
+    console.log('[SSH] dir is not defined');
+    return;
+  }
   if (existsSync(dir)) {
     console.log(`[SSH] ${dir} dir exist`);
     return;
diff --git a/src/remoteCmd.js b/src/remoteCmd.js
index 81f3c19..f274421 100644
--- a/src/remoteCmd.js
+++ b/src/remoteCmd.js
@@ -1,15 +1,14 @@
-const { join } = require('path');
 const { exec } = require('child_process');
 
 const { sshServer, githubWorkspace } = require('./inputs');
 const { writeToFile } = require('./helpers');
 
-const remoteCmd = (cmd, label) => {
-  const localScriptPath = join(githubWorkspace, `local_ssh_script-${label}.sh`);
+const remoteCmd = (content, label) => {
+  const filename = `local_ssh_script-${label}.sh`;
   try {
-    writeToFile(localScriptPath, cmd);
+    writeToFile({ dir: githubWorkspace, filename, content });
 
-    exec(`ssh ${sshServer} 'bash -s' < ${localScriptPath}`, (err, data, stderr) => {
+    exec(`ssh ${sshServer} 'bash -s' < ${filename}`, (err, data, stderr) => {
       if (err) {
         console.log('⚠️ [CMD] Remote script failed. ', err.message);
       } else {
diff --git a/src/sshKey.js b/src/sshKey.js
index 5f5e79e..4d4a16d 100644
--- a/src/sshKey.js
+++ b/src/sshKey.js
@@ -10,7 +10,7 @@ const addSshKey = (content, filename) => {
   writeToFile({ dir, filename: 'known_hosts', content: '' });
   writeToFile({ dir, filename, content, isRequired: true });
 
-  console.log('✅ Ssh key added to `.ssh` dir ', dir);
+  console.log('✅ [SSH] key added to `.ssh` dir ', dir);
 
   return filePath;
 };