X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=PWG3%2Fdielectron%2FAliAnalysisTaskDielectronEfficiency.cxx;h=97ed8d016794a0584fc9907aceb8a11f8996c3a5;hp=ed8e58c410dfed92f334552078d85de113d6fe74;hb=37e9382dd0d8fb31c2b002efc56602dced542446;hpb=2ba5f106e99855c15130395b906f0f25dc9f09a2

diff --git a/PWG3/dielectron/AliAnalysisTaskDielectronEfficiency.cxx b/PWG3/dielectron/AliAnalysisTaskDielectronEfficiency.cxx
index ed8e58c410d..97ed8d01679 100644
--- a/PWG3/dielectron/AliAnalysisTaskDielectronEfficiency.cxx
+++ b/PWG3/dielectron/AliAnalysisTaskDielectronEfficiency.cxx
@@ -359,8 +359,8 @@ void AliAnalysisTaskDielectronEfficiency::FillPlots(AliVEvent *event)
             fHist->Fill("DataCuts","dndyPt",v.Rapidity(),jpsi.GetPt());
 
             //Additional TRD cuts
-            if ( (trackN->GetStatus()&AliESDtrack::kTRDrefit!=0) &&trackN->GetTRDntrackletsPID()>4 ){
-              if ( (trackP->GetStatus()&AliESDtrack::kTRDrefit!=0) &&trackP->GetTRDntrackletsPID()>4 ){
+            if ( ((trackN->GetStatus()&AliESDtrack::kTRDrefit)!=0) && trackN->GetTRDntrackletsPID()>4 ){
+              if ( ((trackP->GetStatus()&AliESDtrack::kTRDrefit)!=0) && trackP->GetTRDntrackletsPID()>4 ){
                 if (motherN){
                   fHist->Fill("DataTRDCuts","JpsiMCPt",motherN->Pt());
                   fHist->Fill("DataTRDCuts","e+Pt",pP->Pt());
@@ -411,8 +411,8 @@ void AliAnalysisTaskDielectronEfficiency::FillMCInfo(AliStack *pStack)
   
   TLorentzVector v;
   //Fill MC info
-  for (Int_t ipart=0; ipart<pStack->GetNprimary(); ++ipart){
-    TParticle *part=pStack->Particle(pStack->GetPrimary(ipart));
+  for (Int_t ipart=0; ipart<pStack->GetNtrack(); ++ipart){
+    TParticle *part=pStack->Particle(ipart);
 //     printf("Particle %d\n",part->GetPdgCode());
     if (part->GetPdgCode()!=fIdMCMother || part->GetNDaughters()!=2) continue;
     TParticle *d1=pStack->Particle(part->GetFirstDaughter());