diff --git a/addon/components/menu/button.hbs b/addon/components/menu/button.hbs
index 2e452c0..814d015 100644
--- a/addon/components/menu/button.hbs
+++ b/addon/components/menu/button.hbs
@@ -1,6 +1,5 @@
 {{! template-lint-disable no-down-event-binding }}
 <button
-  data-test-headlessui-menu-button
   type='button'
   aria-haspopup={{true}}
   aria-controls={{if @isOpen @itemsGuid}}
diff --git a/tests/integration/components/menu-test.js b/tests/integration/components/menu-test.js
index d6f49e3..11b04ba 100644
--- a/tests/integration/components/menu-test.js
+++ b/tests/integration/components/menu-test.js
@@ -24,7 +24,7 @@ function getMenu() {
 }
 
 function getMenuButton() {
-  return find('button,[role="button"],[id^="headlessui-menu-button-"]');
+  return find('button,[role="button"],[data-test-menu-button]');
 }
 
 function getMenuItems() {