Update PlanWriterFlow.py

PlanWriterFlow.py

@@ -56,7 +56,9 @@ class PlanWriterFlow(ContentWriterFlow):
             keys_to_fetch_from_state = ["memory_files"]
             fetched_state = self._fetch_state_attributes_by_keys(keys=keys_to_fetch_from_state)
             plan_file_location = fetched_state["memory_files"]["plan"]
+            memory_files = fetched_state["memory_files"]
             output_payload["command_args"]["plan_file_location"] = plan_file_location
+            output_payload["command_args"]["memory_files"] = memory_files
             return output_payload
         else:
             return output_payload