diff --git a/builders/python-builder.psm1 b/builders/python-builder.psm1
index 4151379..c2541d3 100644
--- a/builders/python-builder.psm1
+++ b/builders/python-builder.psm1
@@ -49,8 +49,8 @@ class PythonBuilder {
 
         $this.HostedToolcacheLocation = $env:RUNNER_TOOL_CACHE
         $this.TempFolderLocation = $env:RUNNER_TEMP
-        $this.WorkFolderLocation = Join-Path $env:RUNNER_TEMP "work"
-        $this.ArtifactFolderLocation = Join-Path $env:RUNNER_TEMP "artifact"
+        $this.WorkFolderLocation =  $workDirectory
+        $this.ArtifactFolderLocation = $artifactDirectory
 
         $this.Version = $version
         $this.Architecture = $architecture