diff --git a/src/bayesvalidrox/surrogate_models/sequential_design.py b/src/bayesvalidrox/surrogate_models/sequential_design.py
index f1b765b7664358a79d077efcc939c2113e789d75..a179bf699a047f365436821a1099d6ad7fc3b15f 100644
--- a/src/bayesvalidrox/surrogate_models/sequential_design.py
+++ b/src/bayesvalidrox/surrogate_models/sequential_design.py
@@ -217,7 +217,7 @@ class SeqDesign():
                     Ynew, _ = Model.run_model_parallel(
                         Xnew, prevRun_No=total_n_samples
                         )
-
+                    total_n_samples += Xnew.shape[0]
                     # ------ Plot the surrogate model vs Origninal Model ------
                     if hasattr(PCEModel, 'adapt_verbose') and \
                        PCEModel.adapt_verbose:
@@ -1869,10 +1869,13 @@ class SeqDesign():
             idx = len([name for name in os.listdir(newpath) if 'Likelihoods_'
                        in name and os.path.isfile(os.path.join(newpath, name))])
             fig, ax = plt.subplots()
-            sns.kdeplot(np.log(valid_likelihoods[valid_likelihoods > 0]),
-                        shade=True, color="g", label='Ref. Likelihood')
-            sns.kdeplot(np.log(Likelihoods[Likelihoods > 0]), shade=True,
-                        color="b", label='Likelihood with PCE')
+            try:
+                sns.kdeplot(np.log(valid_likelihoods[valid_likelihoods > 0]),
+                            shade=True, color="g", label='Ref. Likelihood')
+                sns.kdeplot(np.log(Likelihoods[Likelihoods > 0]), shade=True,
+                            color="b", label='Likelihood with PCE')
+            except:
+                pass
 
             text = f"Hellinger Dist.={distHellinger:.3f}\n logBME={logBME:.3f}"
             "\n DKL={KLD:.3f}"