clean up code in TblBranch

alphatwirl · Aug 27, 2017 · 222230e · 222230e
1 parent ff396f1
commit 222230e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alphatwirl/heppyresult/TblBranch.py b/alphatwirl/heppyresult/TblBranch.py
@@ -41,10 +41,10 @@ def read(self, component):
         tree = file.Get(self.treeName)
 
         for leaf in tree.GetListOfLeaves():
-            branchName = leaf.GetName()
+            branch_entry = self._read_branch_entry(leaf)
+            branchName = branch_entry['name']
             if not branchName in self.branchDict:
                 self.branchOrder.append(branchName)
-                branch_entry = self._read_branch_entry(leaf)
                 self.branchDict[branchName] = branch_entry
             component_entry = self._read_component_entry(leaf)
             component_entry['name'] = component.name