diff --git a/apps/www/.vitepress/theme/components/ComponentPreview.vue b/apps/www/.vitepress/theme/components/ComponentPreview.vue
index a508de3a..3f2a09ab 100644
--- a/apps/www/.vitepress/theme/components/ComponentPreview.vue
+++ b/apps/www/.vitepress/theme/components/ComponentPreview.vue
@@ -43,7 +43,7 @@ watch([style, codeConfig], async () => {
   catch (err) {
     console.error(err)
   }
-}, { immediate: true })
+}, { immediate: true, deep: true })
 </script>
 
 <template>
diff --git a/apps/www/.vitepress/theme/plugins/codewrapper.ts b/apps/www/.vitepress/theme/plugins/codewrapper.ts
index 1314d44b..ec48570b 100644
--- a/apps/www/.vitepress/theme/plugins/codewrapper.ts
+++ b/apps/www/.vitepress/theme/plugins/codewrapper.ts
@@ -8,7 +8,8 @@ export default function (md: MarkdownRenderer) {
   md.renderer.rules.fence = function (tokens, idx, options, env, self) {
     // Check if this is a code block
     const token = tokens[idx]
-    if (token && token.tag === 'code' && token.info) {
+    const isAllowedExtension = (token.info.includes('vue') || token.info.includes('astro') || token.info.includes('ts'))
+    if (token && token.tag === 'code' && isAllowedExtension) {
       // Wrap the code block in CodeWrapper
       return `<CodeWrapper>${defaultFenceRenderer(tokens, idx, options, env, self)}</CodeWrapper>`
     }
diff --git a/apps/www/tsconfig.json b/apps/www/tsconfig.json
index 2bcbad29..2150c34e 100644
--- a/apps/www/tsconfig.json
+++ b/apps/www/tsconfig.json
@@ -12,6 +12,6 @@
     "outDir": "dist",
     "sourceMap": true
   },
-  "include": ["src", ".vitepress/**/*.vue", "scripts/build-registry.ts", ".vitepress/**/*.mts", ".vitepress/**/*.vue", "src/lib/**/*"],
+  "include": ["src", ".vitepress/**/*.vue", "scripts/build-registry.ts", ".vitepress/**/*.mts", ".vitepress/**/*.ts", "src/lib/**/*"],
   "exclude": ["node_modules"]
 }