diff --git a/src/engine/src/core/workflows/WorkflowExecutor.py b/src/engine/src/core/workflows/WorkflowExecutor.py
index 56fcde7e..73e28927 100644
--- a/src/engine/src/core/workflows/WorkflowExecutor.py
+++ b/src/engine/src/core/workflows/WorkflowExecutor.py
@@ -549,7 +549,7 @@ def _prepare_pipeline(self):
 
         # Persist each arg to files in the pipeline file system
         arg_value_file_repo = self.container.load("ArgValueFileRepository")
-        for key in self.state.ctx.pipeline.args:
+        for key in self.state.ctx.args:
             arg_value_file_repo.save(
                 self.state.ctx.pipeline.args_dir + key,
                 self.state.ctx.pipeline.args[key].value