diff --git a/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml b/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml
index a28523de0e..59db0043a7 100644
--- a/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml
+++ b/manifests/v2/base/runtimes/pre-training/torch-distributed.yaml
@@ -31,3 +31,9 @@ spec:
                           env | grep PET_
 
                           pip list
+                      volumeMounts:
+                        - name: workspace
+                          mountPath: /workspace
+                  volumes:
+                    - name: workspace
+                      emptyDir: {}
diff --git a/sdk_v2/kubeflow/training/utils/utils.py b/sdk_v2/kubeflow/training/utils/utils.py
index cc21b96aa7..b999d5c668 100644
--- a/sdk_v2/kubeflow/training/utils/utils.py
+++ b/sdk_v2/kubeflow/training/utils/utils.py
@@ -162,7 +162,6 @@ def get_args_using_train_func(
     # That allows to execute the training script with the entrypoint.
     exec_script = textwrap.dedent(
         """
-                program_path=$(mktemp -d)
                 read -r -d '' SCRIPT << EOM\n
                 {func_code}
                 EOM