diff --git a/js/id/svg/midpoints.js b/js/id/svg/midpoints.js
index c59ef0bca..96048b1ed 100644
--- a/js/id/svg/midpoints.js
+++ b/js/id/svg/midpoints.js
@@ -2,8 +2,8 @@ iD.svg.Midpoints = function(projection) {
     return function drawMidpoints(surface, graph, entities, filter) {
         var midpoints = {};
 
-        if (!surface.select('.layer-hit').select('g.vertex').node()) {
-            return surface.select('.layer-hit').selectAll('g.midpoint').remove();
+        if (!surface.select('.layer-hit.g.vertex').node()) {
+            return surface.select('.layer-hit.g.midpoint').remove();
         }
 
         for (var i = 0; i < entities.length; i++) {