diff --git a/javascripts/discourse/api-initializers/table-editor.js b/javascripts/discourse/api-initializers/table-editor.js
index b3e6e0c..76c1ea7 100644
--- a/javascripts/discourse/api-initializers/table-editor.js
+++ b/javascripts/discourse/api-initializers/table-editor.js
@@ -16,7 +16,6 @@ export default apiInitializer("0.11.1", (api) => {
     const openPopupBtn = document.createElement("button");
     openPopupBtn.classList.add(
       "open-popup-link",
-      "btn-edit-md-table",
       "btn-default",
       "btn",
       "btn-icon-text"
diff --git a/test/acceptance/table-builder-test.js b/test/acceptance/table-builder-test.js
index 91a928b..42d015e 100644
--- a/test/acceptance/table-builder-test.js
+++ b/test/acceptance/table-builder-test.js
@@ -64,12 +64,9 @@ acceptance("Table Builder - Edit Table", function (needs) {
   test("Can see edit button on post with table", async function (assert) {
     await visit("/t/960");
     await focus(document.querySelector(".cooked .md-table"));
-    assert.ok(exists("button.btn-edit-md-table"), "Edit Table button exists");
-    assert.ok(
-      visible("button.btn-edit-md-table"),
-      "Edit Table button is visible"
-    );
-    await click("button.btn-edit-md-table");
+    assert.ok(exists("button.btn-edit-table"), "Edit Table button exists");
+    assert.ok(visible("button.btn-edit-table"), "Edit Table button is visible");
+    await click("button.btn-edit-table");
     assert.ok(
       exists(".insert-table-modal-modal"),
       "Table Builder modal exists"