diff --git a/notebooks/fidelity.qmd b/notebooks/fidelity.qmd
index 7a57cc0b6562cacf0e1189508a262a67db50cbc7..2b7e384ace148025834ba79617de24535c9959ec 100644
--- a/notebooks/fidelity.qmd
+++ b/notebooks/fidelity.qmd
@@ -27,7 +27,7 @@ x = select_factual(counterfactual_data, chosen)
 
 
 ```{julia}
-niter = 10
+niter = 10000
 nsamples = 100
 plts = []
 for target in ce.data.y_levels
diff --git a/src/sampling.jl b/src/sampling.jl
index 2cf346ca70ac22e307080bf1f06c0e44738e8396..8a6625ee9d1b02ba37b8263bf3772d2ee9c6e0ce 100644
--- a/src/sampling.jl
+++ b/src/sampling.jl
@@ -13,7 +13,7 @@ end
 
 function EnergySampler(
     ce::CounterfactualExplanation;
-    opt::JointEnergyModels.AbstractSamplingRule=ImproperSGLD(),
+    opt::JointEnergyModels.AbstractSamplingRule=SGLD(),
     niter::Int=100,
     nsamples::Int=1000
 )