diff --git a/modules/core/tree.js b/modules/core/tree.js
index 179cc7d1b..b95185c66 100644
--- a/modules/core/tree.js
+++ b/modules/core/tree.js
@@ -98,7 +98,7 @@ export function coreTree(head) {
         }
 
         return rtree.search(extent.bbox())
-            .map(function(bbox) { return head.entity(bbox.id); });
+            .map(function(bbox) { return graph.entity(bbox.id); });
     };