Update of hfe code

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEspectrum.cxx

diff --git a/PWGHF/hfe/AliHFEspectrum.cxx b/PWGHF/hfe/AliHFEspectrum.cxx
index 4f032a43ce2cdc6f71170c78a1b0bb9af3440b90..3349817966cf39326bd79d9cfa7bab9845a25fae 100644 (file)
--- a/PWGHF/hfe/AliHFEspectrum.cxx
+++ b/PWGHF/hfe/AliHFEspectrum.cxx
@@ -390,7 +390,10 @@ Bool_t AliHFEspectrum::Init(const AliHFEcontainer *datahfecontainer, const AliHF
         ptcorrelation->Draw("colz");
       }
     }
-    if(fWriteToFile) ccontaminationspectrum->SaveAs("contaminationspectrum.eps");
+    if(fWriteToFile){ 
+      ccontaminationspectrum->SaveAs("contaminationspectrum.eps");
+      ccorrelation->SaveAs("correlationMatrix.eps");
+    }
   }
 
   TFile *file = TFile::Open("tests.root","recreate");