diff --git a/cli/utils/templates.mjs b/cli/utils/templates.mjs
index cb68dce2..0de57453 100644
--- a/cli/utils/templates.mjs
+++ b/cli/utils/templates.mjs
@@ -25,6 +25,7 @@ const component = ({ name }) => {
contents: `
-
+ <${upperName}Root v-bind="rootProps" :class="ui.root({ class: props.class })" />
`
}
@@ -66,7 +73,7 @@ const theme = ({ name }) => {
return {
filename: `src/theme/${camelName}.ts`,
contents: `
-export default {
+export default (config: { colors: string[] }) => ({
slots: {
root: ''
},
@@ -76,7 +83,7 @@ export default {
defaultVariants: {
}
-}
+})
`
}
}