diff --git a/sdk_v2/kubeflow/training/utils/utils.py b/sdk_v2/kubeflow/training/utils/utils.py
index cc21b96aa7..ee6aea3c20 100644
--- a/sdk_v2/kubeflow/training/utils/utils.py
+++ b/sdk_v2/kubeflow/training/utils/utils.py
@@ -166,8 +166,8 @@ def get_args_using_train_func(
                 read -r -d '' SCRIPT << EOM\n
                 {func_code}
                 EOM
-                printf "%s" \"$SCRIPT\" > \"{func_file}\"
-                {entrypoint} \"{func_file}\""""
+                printf "%s" \"$SCRIPT\" > \"$program_path\"/\"{func_file}\"
+                {entrypoint} \"$program_path\"/\"{func_file}\""""
     )
 
     # Add function code to the execute script.