Bring back update logic as it is in fact required.

d3 · May 25, 2021 · e0b61ec · e0b61ec
1 parent e47584c
commit e0b61ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/selection/join.js b/src/selection/join.js
@@ -6,7 +6,10 @@ function(onenter, onupdate, onexit) {
   } else {
     enter = enter.append(onenter + "");
   }
-  if (onupdate != null) update = onupdate(update);
+  if (onupdate != null) {
+    update = onupdate(update);
+    if (update) update = update.selection();
+  }
   if (onexit == null) exit.remove(); else onexit(exit);
   return enter && update ? enter.merge(update).order() : update;
 }