diff --git a/streamlit/sf_conn.py b/streamlit/sf_conn.py
index 31ceac5..10b30e4 100644
--- a/streamlit/sf_conn.py
+++ b/streamlit/sf_conn.py
@@ -64,14 +64,14 @@ def save_to_sf(dag_name, **kwargs):
     # attempt_logs_file = "attempt_logs_sample.csv"
     # conf = "{\"" + "training_results_file_name" + "\":\"" + {training_results_file} + "\", \"" + "attempt_logs_file_name" + "\":\"" + {attempt_logs_file} + "\"}".format(training_results_file=training_results_file, attempt_logs_file=attempt_logs_file)
 
-    conf = f"""{{
-        "{list(kwargs.keys())[0]}": "{list(kwargs.values())[0]}",
-        "{list(kwargs.keys())[1]}": "{list(kwargs.values())[1]}"
-        }}"""
+    # conf = f"""{{
+    #     "{list(kwargs.keys())[0]}": "{list(kwargs.values())[0]}",
+    #     "{list(kwargs.keys())[1]}": "{list(kwargs.values())[1]}"
+    #     }}"""
 
-    # conf = f"""
-    # {{ {json.dumps(kwargs)} 
-    # }}"""
+    conf = f"""
+    {{ {json.dumps(kwargs)} 
+    }}"""
 
     payload = mwaa_cli_command + " " + dag_name + " --conf '{}'".format(conf)
     headers = {