diff --git a/js/id/behavior/draw_way.js b/js/id/behavior/draw_way.js
index f724613da..3089bd6ee 100644
--- a/js/id/behavior/draw_way.js
+++ b/js/id/behavior/draw_way.js
@@ -21,7 +21,7 @@ iD.behavior.DrawWay = function(context, wayId, index, mode, baseGraph) {
         } else if (datum.type === 'midpoint' || datum.type === 'way') {
             var way = datum.type === 'way' ?
                 datum :
-                baseGraph.entity(datum.ways[0].id);
+                context.entity(datum.ways[0].id);
             loc = iD.geo.chooseIndex(way, d3.mouse(context.surface().node()), context).loc;
         }