Fix (Andrei)

diff --git a/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx b/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
index f370e864906efc2851c49bea31ebb15ee596b5ef..9c43b0ca570189f41afc77f185c13c583c32a425 100644 (file)
--- a/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
+++ b/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
@@ -31,6 +31,7 @@
 
 #include "AliVEvent.h"
 #include "AliAODEvent.h"
+#include "AliAODHandler.h"
 #include "AliESDEvent.h"
 #include "AliAnalysisVertexingHF.h"
 #include "AliAnalysisTaskSE.h"
@@ -221,6 +222,11 @@ void AliAnalysisTaskSEVertexingHF::UserExec(Option_t */*option*/)
   // event in memory rather than the input (ESD) event. (A.G. 27/04/09)
   if (AODEvent() && IsStandardAOD()) event = dynamic_cast<AliVEvent*> (AODEvent());
 
+  if (dynamic_cast<AliAODEvent*>(event)) {
+     AliAODHandler *aodhandler = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
+     if (aodhandler) aodhandler->SetFillExtension(kTRUE);
+  }   
+
   // heavy flavor vertexing
   fVHF->FindCandidates(event,
                       fVerticesHFTClArr,