Fix needed to connect friends to trees and chains.

[u/mrichter/AliRoot.git] / STEER / AliAODInputHandler.cxx

diff --git a/STEER/AliAODInputHandler.cxx b/STEER/AliAODInputHandler.cxx
index fce9c9721567dcbab323e70f856e61f5cc50ac09..66794838023b10c132434d1e7b5d2e4f51376d12 100644 (file)
--- a/STEER/AliAODInputHandler.cxx
+++ b/STEER/AliAODInputHandler.cxx
@@ -61,8 +61,13 @@ Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /*opt*/)
     fTree = tree;
     TIter next(fFriends);
     TNamed* obj;
+    
     while(obj = (TNamed*)next()) {
-       fTree->AddFriend("aodTree", obj->GetName());
+       if (fTree->GetTree()) {
+           (fTree->GetTree())->AddFriend("aodTree", obj->GetName());
+       } else {
+           fTree->AddFriend("aodTree", obj->GetName());
+       }
     }
     
     if (!fTree) return kFALSE;