diff --git a/src/components/GenerateMenuButton.test.ts b/src/components/GenerateMenuButton.test.ts
index 2acf6db..33bfc15 100644
--- a/src/components/GenerateMenuButton.test.ts
+++ b/src/components/GenerateMenuButton.test.ts
@@ -54,7 +54,7 @@ describe('GenerateMenuButton', () => {
await wrapper.get('button[data-testid="generate-menu-toggle"]').trigger('click')
expect(wrapper.find('[data-testid="generate"]').exists()).toBe(true)
- await wrapper.get('div.generate-menu').trigger('keydown', { key: 'Escape' })
+ await wrapper.get('div.toolbar-menu').trigger('keydown', { key: 'Escape' })
expect(wrapper.find('[data-testid="generate"]').exists()).toBe(false)
wrapper.unmount()
diff --git a/src/components/GenerateMenuButton.vue b/src/components/GenerateMenuButton.vue
index c516cc9..a740c80 100644
--- a/src/components/GenerateMenuButton.vue
+++ b/src/components/GenerateMenuButton.vue
@@ -1,68 +1,39 @@
-
+
+