diff --git a/openavmkit/benchmark.py b/openavmkit/benchmark.py
index 348d938..2f1f73b 100644
--- a/openavmkit/benchmark.py
+++ b/openavmkit/benchmark.py
@@ -3590,7 +3590,7 @@ def _run_hedonic_models(
     )
 
     if run_ensemble:
-        _perform_ensemble(
+        ensemble_results = _perform_ensemble(
             df_sales=df_sales,
             df_universe=df_universe,
             model_group=model_group,
@@ -4698,4 +4698,4 @@ def _quick_shap(
     shaps = _calc_shap(smr.model, X_train, X_train)
 
     if plot:
-        plot_full_beeswarm(shaps, title=title)
\ No newline at end of file
+        plot_full_beeswarm(shaps, title=title)