1 files changed, 3 insertions, 2 deletions
diff --git a/src/stratified_sampling.hpp b/src/stratified_sampling.hpp
index ed53b6d..c04369e 100644
--- a/src/stratified_sampling.hpp
+++ b/src/stratified_sampling.hpp
@@ -141,8 +141,9 @@ void stratified_sampling<L>::draw(int N) {
         m=0;
         s=0;
         for(int j=0;j<M[i];j++){
-            m=m+X[i]();
-            s=s+X[i].current()*X[i].current();
+            double temp = X[i]();
+            m=m+temp;
+            s=s+temp*temp;
         }
         oldmean=mean[i];
         mean[i]=(mean[i]*cumM[i]+m)/(cumM[i]+M[i]);