diff --git a/scripts/templates/task_input_config.2pc_balanced_psi_dp.json b/scripts/templates/task_input_config.2pc_balanced_psi_dp.json
index 4b4d98d3..9dba3fe5 100644
--- a/scripts/templates/task_input_config.2pc_balanced_psi_dp.json
+++ b/scripts/templates/task_input_config.2pc_balanced_psi_dp.json
@@ -28,7 +28,7 @@
     "domain": "data_prep",
     "name": "psi",
     "version": "0.0.4",
-    "attr_paths": ["input/receiver_input/key", "input/sender_input/key", "protocol", "precheck_input", "bucket_size", "curve_type","left_side"],
+    "attr_paths": ["input/receiver_input/key", "input/sender_input/key", "protocol", "precheck_input", "bucket_size", "ecdh_curve","left_side"],
     "attrs": [{
       "ss": ["id1"]
     }, {
@@ -49,4 +49,4 @@
   "sf_input_ids": ["alice-dp-table","bob-dp-table"],
   "sf_output_ids": ["psi-dp-output"],
   "sf_output_uris": ["psi-dp-output-dp.csv"]
-}
\ No newline at end of file
+}