From c35fc1b7445701cb3ffa664cfb8a9d088f59c273 Mon Sep 17 00:00:00 2001
From: David Hadley <d.r.hadley@warwick.ac.uk>
Date: Thu, 18 Oct 2018 10:19:26 +0100
Subject: [PATCH] Fix for Bug ROOT-9731.

---
 tree/dataframe/src/RDFInterfaceUtils.cxx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tree/dataframe/src/RDFInterfaceUtils.cxx b/tree/dataframe/src/RDFInterfaceUtils.cxx
index 889b0d5..a77884b 100644
--- a/tree/dataframe/src/RDFInterfaceUtils.cxx
+++ b/tree/dataframe/src/RDFInterfaceUtils.cxx
@@ -19,6 +19,7 @@
 #include <TString.h>
 #include <TTree.h>
 #include <TBranchElement.h>
+#include <TBranchObject.h>
 
 #include <iosfwd>
 #include <stdexcept>
@@ -98,7 +99,7 @@ void GetBranchNamesImpl(TTree &t, std::set<std::string> &bNamesReg, ColumnNames_
       for (auto b : *branches) {
          TBranch *branch = static_cast<TBranch *>(b);
          auto branchName = std::string(branch->GetName());
-         if (branch->IsA() == TBranch::Class()) {
+         if (branch->IsA() == TBranch::Class() || branch->IsA() == TBranchObject::Class()) {
             // Leaf list
 
             auto listOfLeaves = branch->GetListOfLeaves();
-- 
2.7.4