diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py
index c30385862..d087d53d6 100644
--- a/baybe/insights/shap.py
+++ b/baybe/insights/shap.py
@@ -224,9 +224,11 @@ def from_recommender(
             searchspace, objective, measurements
         )
 
+        data = measurements[[p.name for p in searchspace.parameters]]
+
         return cls.from_surrogate(
             surrogate_model,
-            searchspace.transform(measurements) if use_comp_rep else measurements,
+            searchspace.transform(data) if use_comp_rep else data,
             explainer_cls=explainer_cls,
             use_comp_rep=use_comp_rep,
         )