diff --git a/tests/components/input_shortcut.test.ts b/tests/components/input_shortcut.test.ts
index 55b6625..748d52a 100644
--- a/tests/components/input_shortcut.test.ts
+++ b/tests/components/input_shortcut.test.ts
@@ -1,8 +1,10 @@
 
-import { vi, expect, test, beforeEach } from 'vitest'
-import { mount, VueWrapper } from '@vue/test-utils'
+import { vi, expect, test, beforeEach, afterAll } from 'vitest'
+import { mount, VueWrapper, enableAutoUnmount } from '@vue/test-utils'
 import InputShortcut from '../../src/components/InputShortcut.vue'
 
+enableAutoUnmount(afterAll)
+
 let wrapper: VueWrapper<any>
 
 window.api = {
diff --git a/tests/screens/commands.test.ts b/tests/screens/commands.test.ts
index 1fbee2b..897f748 100644
--- a/tests/screens/commands.test.ts
+++ b/tests/screens/commands.test.ts
@@ -1,9 +1,11 @@
 
-import { vi, beforeEach, expect, test } from 'vitest'
-import { mount } from '@vue/test-utils'
+import { vi, beforeEach, expect, test, afterAll } from 'vitest'
+import { enableAutoUnmount, mount } from '@vue/test-utils'
 import Commands from '../../src/screens/Commands.vue'
 import defaults from '../../defaults/settings.json'
 
+enableAutoUnmount(afterAll)
+
 window.api = {
   config: {
     load: vi.fn(() => defaults),
diff --git a/tests/screens/main.test.ts b/tests/screens/main.test.ts
index 46d4033..2b063f3 100644
--- a/tests/screens/main.test.ts
+++ b/tests/screens/main.test.ts
@@ -1,6 +1,6 @@
 
-import { vi, beforeEach, expect, test } from 'vitest'
-import { mount } from '@vue/test-utils'
+import { vi, beforeEach, expect, test, afterAll } from 'vitest'
+import { enableAutoUnmount, mount } from '@vue/test-utils'
 import { store } from '../../src/services/store'
 import Main from '../../src/screens/Main.vue'
 import Sidebar from '../../src/components/Sidebar.vue'
@@ -12,6 +12,8 @@ import * as _Assistant from '../../src/services/assistant'
 import useEventBus  from '../../src/composables/useEventBus'
 const { emitEvent } = useEventBus()
 
+enableAutoUnmount(afterAll)
+
 window.api = {
   config: {
     load: vi.fn(() => defaults),
diff --git a/tests/screens/prompt.test.ts b/tests/screens/prompt.test.ts
index 26b97a8..9ff4de7 100644
--- a/tests/screens/prompt.test.ts
+++ b/tests/screens/prompt.test.ts
@@ -1,6 +1,6 @@
 
-import { vi, beforeEach, expect, test } from 'vitest'
-import { mount } from '@vue/test-utils'
+import { vi, beforeEach, expect, test, afterAll } from 'vitest'
+import { enableAutoUnmount, mount } from '@vue/test-utils'
 import Prompt from '../../src/components/Prompt.vue'
 import PromptAnywhere from '../../src/screens/PromptAnywhere.vue'
 import defaults from '../../defaults/settings.json'
@@ -8,6 +8,8 @@ import defaults from '../../defaults/settings.json'
 import useEventBus  from '../../src/composables/useEventBus'
 const { emitEvent } = useEventBus()
 
+enableAutoUnmount(afterAll)
+
 window.api = {
   config: {
     load: vi.fn(() => defaults),
diff --git a/tests/screens/wait.test.ts b/tests/screens/wait.test.ts
index bfdb16b..9b18e4f 100644
--- a/tests/screens/wait.test.ts
+++ b/tests/screens/wait.test.ts
@@ -1,8 +1,10 @@
 
-import { vi, expect, test } from 'vitest'
-import { mount } from '@vue/test-utils'
+import { vi, expect, test, afterAll } from 'vitest'
+import { enableAutoUnmount, mount } from '@vue/test-utils'
 import Wait from '../../src/screens/Wait.vue'
 
+enableAutoUnmount(afterAll)
+
 window.api = {
   commands: {
     cancel: vi.fn()