diff --git a/components/button/button.tsx b/components/button/button.tsx
index 8419368..4115ff8 100644
--- a/components/button/button.tsx
+++ b/components/button/button.tsx
@@ -1,7 +1,7 @@
 import { computed, defineComponent } from 'vue'
 import { useProviderConfigState } from '../config-provider/context'
 import useStyle from './style'
-export default defineComponent({
+const Button = defineComponent({
   name: 'AButton',
   inheritAttrs: false,
   props: {
@@ -9,7 +9,7 @@ export default defineComponent({
       type: String
     }
   },
-  setup(props, { slots }) {
+  setup(props, { slots, attrs }) {
     const { getPrefixCls } = useProviderConfigState()
     const prefixCls = computed(() => getPrefixCls('btn', props.prefixCls))
     const [wrapSSR, hashId] = useStyle(prefixCls)
@@ -18,7 +18,9 @@ export default defineComponent({
         [prefixCls.value]: true,
         [hashId.value]: true
       }
-      return wrapSSR(<button class={cls}>{slots.default?.()}</button>)
+      return wrapSSR(<button class={[cls, attrs.class]}>{slots.default?.()}</button>)
     }
   }
 })
+
+export default Button
diff --git a/components/theme/internal.ts b/components/theme/internal.ts
index 4aaed50..c819b2e 100644
--- a/components/theme/internal.ts
+++ b/components/theme/internal.ts
@@ -77,6 +77,6 @@ export function useToken(): [ComputedRef<Theme<SeedToken, MapToken>>, ComputedRe
   return [mergedTheme, computed(() => cacheToken.value?.[0]), computed(() => (designTokenContext.hashed ? cacheToken.value?.[1] : ''))]
 }
 
-export type UseComponentStyleResult = [(node: VNodeChild) => VNodeChild, string]
+export type UseComponentStyleResult = [(node: VNodeChild) => VNodeChild, ComputedRef<string>]
 
 export type GenerateStyle<ComponentToken extends object = AliasToken, ReturnType = CSSInterpolation> = (token: ComponentToken) => ReturnType
diff --git a/components/theme/util/genComponentStyleHook.ts b/components/theme/util/genComponentStyleHook.ts
index 5033e52..1cf25e0 100644
--- a/components/theme/util/genComponentStyleHook.ts
+++ b/components/theme/util/genComponentStyleHook.ts
@@ -92,7 +92,7 @@ export default function genComponentStyleHook<ComponentName extends OverrideComp
         flush(component, mergedComponentToken)
         return [genCommonStyle(token.value, prefixCls.value), styleInterpolation]
       }),
-      hashId.value
+      hashId
     ]
   }
 }
diff --git a/site/.vitepress/theme/index.ts b/site/.vitepress/theme/index.ts
index 486972a..7063c2e 100644
--- a/site/.vitepress/theme/index.ts
+++ b/site/.vitepress/theme/index.ts
@@ -2,11 +2,13 @@ import type { Theme } from 'vitepress'
 // eslint-disable-next-line import/no-named-as-default
 import DefaultTheme from 'vitepress/theme'
 import { AntdTheme } from 'vite-plugin-vitepress-demo/theme'
+import Antd from '../../../components'
 
 export default {
   ...DefaultTheme,
   enhanceApp(ctx) {
     DefaultTheme.enhanceApp?.(ctx)
     ctx.app.component('Demo', AntdTheme)
+    ctx.app.use(Antd)
   }
 } as Theme
diff --git a/site/demos/button/basic.vue b/site/demos/button/basic.vue
index 89b66b1..bbfa221 100644
--- a/site/demos/button/basic.vue
+++ b/site/demos/button/basic.vue
@@ -10,7 +10,9 @@ title: 基础按钮
 <script lang="ts" setup></script>
 
 <template>
-  <div></div>
+  <div>
+    <a-button>这是按钮</a-button>
+  </div>
 </template>
 
 <style scoped></style>