diff --git a/examples/bat_gen/runBatFit.cxx b/examples/bat_gen/runBatFit.cxx
index bddbb6dd..ce807297 100644
--- a/examples/bat_gen/runBatFit.cxx
+++ b/examples/bat_gen/runBatFit.cxx
@@ -25,7 +25,7 @@
 #include <BAT/BCLog.h>
 #include <BAT/BCParameterSet.h>
 
-#include <TCanvas.h>
+//#include <TCanvas.h>
 #include <TFile.h>
 #include <TTree.h>
 
diff --git a/src/DecayTreeVectorIntegral.cxx b/src/DecayTreeVectorIntegral.cxx
index 43c96446..2be19ee1 100644
--- a/src/DecayTreeVectorIntegral.cxx
+++ b/src/DecayTreeVectorIntegral.cxx
@@ -16,6 +16,9 @@ DecayTreeVectorIntegral::DecayTreeVectorIntegral(const DecayTreeVector& dtv)
       Diagonals_(DecayTrees_.size()),
       OffDiagonals_(DecayTrees_.size() - 1)
 {
+    for (size_t i = 0; i < Diagonals_.size(); ++i)
+        Diagonals_[i].value() = 1.;
+
     for (size_t i = 0; i < OffDiagonals_.size(); ++i)
         OffDiagonals_[i] = ComplexIntegralElementMatrix::value_type(DecayTrees_.size() - i - 1);
 }