diff --git a/toolkit.py b/toolkit.py
index 0688c124568aec00fb37978ec6cb9957b6f43813..598d60e18cc991759288994462c5ee603406d769 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -887,9 +887,9 @@ class ClassificationProject(object):
             logger.info("Evaluating score for entry {}/{}".format(start, entries))
             logger.debug("Loading next batch")
             x_from_tree = tree2array(tree,
-                                     branches=self.fields+self.identifiers,
+                                     branches=self.branches+self.identifiers,
                                      start=start, stop=start+batch_size)
-            x_eval = rec2array(x_from_tree[self.fields])
+            x_eval = rec2array(x_from_tree[self.branches])
 
             if len(self.identifiers) > 0:
                 # create list of booleans that indicate which events where used for training