diff --git a/PeakOTron.py b/PeakOTron.py
index 9d315483340802ac8146b3181ee2e6eb359d3104..56fa80019168634def97f981991d000d00af4f41 100644
--- a/PeakOTron.py
+++ b/PeakOTron.py
@@ -410,9 +410,7 @@ class PeakOTron:
             y_hat_all = self.GetModel(x_all, prefit=prefit, bin_units=True)
         else:
             y_hat_all = DRM(x_all, **pars)
-            
-        y_hat_err_all = np.sqrt(y_hat_all)
-        
+     
         conds = []
         
         cond_count = (y_all>1)
@@ -442,7 +440,7 @@ class PeakOTron:
         y = y_all[conds]
         y_err = y_err_all[conds]
         y_hat = y_hat_all[conds]
-        y_hat_err =  y_hat_err_all[conds]
+        y_hat_err =  np.sqrt(y_hat_all[conds])