diff --git a/lenstronomy/Sampling/sampler.py b/lenstronomy/Sampling/sampler.py
index <HASH>..<HASH> 100644
--- a/lenstronomy/Sampling/sampler.py
+++ b/lenstronomy/Sampling/sampler.py
@@ -109,7 +109,7 @@ class Sampler(object):
 
         sampler.run_mcmc(p0, n_burn + n_run, progress=progress)
         flat_samples = sampler.get_chain(discard=n_burn, thin=1, flat=True)
-        dist = sampler.get_log_prob(flat=True)
+        dist = sampler.get_log_prob(flat=True, discard=n_burn, thin=1)
         if is_master_pool:
             print('Computing the MCMC...')
             print('Number of walkers = ', n_walkers)
bug fix length of the logL chain