diff --git a/experiments/setup.jl b/experiments/setup.jl
index 738d016ff3f3083d69fd3d1faf6f0a6bb95fdb4d..476df92079e4eed4704dd13588b22f0144f93522 100644
--- a/experiments/setup.jl
+++ b/experiments/setup.jl
@@ -57,8 +57,8 @@ function run_experiment(counterfactual_data::CounterfactualData, test_data::Coun
     # Data
     X, labels, n_obs, save_name, batch_size, sampler = prepare_data(
         counterfactual_data;
-        𝒟x=𝒟x,
-        sampling_batch_size=sampling_batch_size,
+        𝒟x=params.𝒟x,
+        sampling_batch_size=params.sampling_batch_size,
     )
 
     # MODELS ----------