diff --git a/alpha_automl/pipeline_synthesis/setup_search.py b/alpha_automl/pipeline_synthesis/setup_search.py
index 4f092607..c654c5cc 100644
--- a/alpha_automl/pipeline_synthesis/setup_search.py
+++ b/alpha_automl/pipeline_synthesis/setup_search.py
@@ -104,7 +104,7 @@ def update_config(task_name, metric, grammar, metadata):
     config['METRIC'] = metric
     config['DATASET'] = f'DATASET_{task_name}'
     config['GRAMMAR'] = grammar
-    metafeatures = compute_metafeatures(metadata)
+    metafeatures = compute_metafeatures(metric, metadata)
     config['DATASET_METAFEATURES'] = metafeatures + [0] * (8 - len(metafeatures))
 
     return config