diff --git a/src/index.ts b/src/index.ts
index b512781..0d9eeab 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -11,7 +11,9 @@ import * as buildtool from "./buildtool";
 async function run() {
     const platform = process.platform;
     const architecture = process.arch;
-    const workspaceDir = core.getInput("working-directory") || process.cwd();
+
+    const workspaceDir = process.cwd();
+    const workspaceExt = core.getInput("working-directory");
 
     const options: buildtool.RunBuildOptions = {
         Tasks: core.getInput("tasks"),
@@ -21,7 +23,7 @@ async function run() {
     const startupOptions = core.getInput("startup-options").split(" ");
 
     const helperScript = await core.group("Generate script", async () => {
-        const helperScript = await matlab.generateScript(workspaceDir, command);
+        const helperScript = await matlab.generateScript(workspaceDir + "/" + workspaceExt, command);
         core.info("Successfully generated script");
         return helperScript;
     });