diff --git a/src/sagemaker/workflow/functions.py b/src/sagemaker/workflow/functions.py
index 53bcc5cc78..730698dbf5 100644
--- a/src/sagemaker/workflow/functions.py
+++ b/src/sagemaker/workflow/functions.py
@@ -32,7 +32,10 @@ class Join(PipelineVariable):
         bucket = ParameterString('bucket', default_value='my-bucket')
 
         TrainingInput(
-            s3_data=Join(on='/', ['s3:/', bucket, ExecutionVariables.PIPELINE_EXECUTION_ID]),
+            s3_data=Join(
+                on='/',
+                values=['s3:/', bucket, ExecutionVariables.PIPELINE_EXECUTION_ID]
+            ),
             content_type="text/csv")
 
     Attributes: