diff --git a/test/components/elements/Button.spec.ts b/test/components/elements/Button.spec.ts
index 123011ce..6dfb5040 100644
--- a/test/components/elements/Button.spec.ts
+++ b/test/components/elements/Button.spec.ts
@@ -1,11 +1,8 @@
-// @vitest-environment nuxt
 import { describe, it, expect } from 'vitest'
-import Button from '../../src/runtime/components/elements/Button.vue'
+import { UButton } from '#components'
 import type { TypeOf } from 'zod'
 import ComponentRender from '../component-render'
 
-type ButtonOptions = TypeOf<typeof Button.props>
-
 describe('Button', () => {
   it.each([
     [ 'basic case', { } ],
@@ -14,12 +11,12 @@ describe('Button', () => {
     [ 'rounded full', { props: { ui: { rounded: 'rounded-full' } } } ],
     [ '<UButton icon="i-heroicons-pencil-square" size="sm" color="primary" square variant="solid" />' ]
     // @ts-ignore
-  ])('renders %s correctly', async (nameOrHtml: string, options: ButtonOptions) => {
+  ])('renders %s correctly', async (nameOrHtml: string, options: TypeOf<typeof Button.props>) => {
     if (options !== undefined) {
       options.slots = options.slots || { default: () => 'label' }
       options.slots.default = options.slots.default || (() => 'label')
     }
-    const html = await ComponentRender(nameOrHtml, options, Button)
+    const html = await ComponentRender(nameOrHtml, options, UButton)
     expect(html).toMatchSnapshot()
   })
 })