diff --git a/src/common/create_component/script.py b/src/common/create_component/script.py
index 87e52fd7..28fa3896 100644
--- a/src/common/create_component/script.py
+++ b/src/common/create_component/script.py
@@ -170,11 +170,10 @@ def set_par_values(config) -> None:
       value = arg.get("default", arg.get("example", "..."))
     elif key == "de_train":
       value = "resources/neurips-2023-kaggle/de_train.parquet"
+    elif key == "de_train":
+      value = "resources/neurips-2023-kaggle/de_train.h5ad"
     elif key == "id_map":
       value = "resources/neurips-2023-kaggle/id_map.csv"
-    elif arg.get("direction", "input") == "input":
-      key_strip = key.replace("input_", "")
-      value = f'resources_test/{par["task"]}/pancreas/{key_strip}.h5ad'
     else:
       key_strip = key.replace("output_", "")
       value = f'{key_strip}.h5ad'