diff --git a/BayesValidRox/BayesInference/BayesInference.py b/BayesValidRox/BayesInference/BayesInference.py
index 78a5d9ffa2896cfdb2a16210675238c049675557..d4a7bd39cd9b22472c68ebf216a5ead7505db4b0 100644
--- a/BayesValidRox/BayesInference/BayesInference.py
+++ b/BayesValidRox/BayesInference/BayesInference.py
@@ -267,8 +267,8 @@ class BayesInference:
                         except:
                             sigma2 = 0.0
 
-                        # covMatrix = np.diag(sigma2 * np.eye(nout))
-                        covMatrix = np.diag(sigma2 * totalSigma2s)
+                        covMatrix += sigma2 * np.eye(nout)
+                        # covMatrix = np.diag(sigma2 * totalSigma2s)
 
                 # Add the std of the PCE is chosen as emulator.
                 if self.emulator:
@@ -653,8 +653,8 @@ class BayesInference:
                             sigma2 = 0.0
 
                         # Convert biasSigma2s to a covMatrix
-                        # cov = np.diag(sigma2 * np.ones(len(pred)))
-                        cov = np.diag(sigma2 * totalSigma2)
+                        cov += sigma2 * np.eye(len(pred))
+                        # cov = np.diag(sigma2 * totalSigma2)
 
                 if self.emulator:
                     stdPCE = PCEModel.RMSE[var] if PCEModel.RMSE is not None else PosteriorPred_std[var][i]
@@ -1027,7 +1027,7 @@ class BayesInference:
                                      quantiles=[0.15, 0.5, 0.85],show_titles=True,
                                      title_fmt=self.Corner_title_fmt,
                                      labelpad=0.2,
-                                     use_math_text = True,
+                                     use_math_text=True,
                                      title_kwargs={"fontsize": 28},
                                      plot_datapoints=False,
                                      plot_density=False,