diff --git a/packages/nbinteract-core/src/manager.js b/packages/nbinteract-core/src/manager.js
index <HASH>..<HASH> 100644
--- a/packages/nbinteract-core/src/manager.js
+++ b/packages/nbinteract-core/src/manager.js
@@ -77,7 +77,7 @@ export class WidgetManager extends HTMLManager {
     const model = await util.msgToModel(msg, this)
     if (model) {
       // Remove all widget buttons
-      util.removeButtons(cell)
+      util.removeButtons()
 
       // Display widget
       const outputEl = util.cellToWidgetOutput(cell)
Remove top level button properly when widgets render

Before, the top-level button would never disappear after widgets render.
Now, it disappears properly.