diff --git a/src/data_processors/transform/config.vsh.yaml b/src/data_processors/transform/config.vsh.yaml
index 91db0477..863207af 100644
--- a/src/data_processors/transform/config.vsh.yaml
+++ b/src/data_processors/transform/config.vsh.yaml
@@ -9,6 +9,10 @@ description: |
     - Reorder the integrated dataset to match the input dataset if needed.
     - Transform the corrected feature output to an embedding.
     - Transform an embedding to a graph output.
+info:
+  test_setup:
+    default:
+      expected_method_types: feature
 resources:
   - type: python_script
     path: script.py
diff --git a/src/workflows/run_benchmark/main.nf b/src/workflows/run_benchmark/main.nf
index 331fc460..35100060 100644
--- a/src/workflows/run_benchmark/main.nf
+++ b/src/workflows/run_benchmark/main.nf
@@ -109,7 +109,7 @@ workflow run_wf {
       toState: { id, output, state, comp ->
         state + [
           method_id: comp.config.name,
-          method_types: comp.config.info.method_types
+          method_types: comp.config.info.method_types,
           method_output: output.output
         ]
       }