diff --git a/src/compile.ts b/src/compile.ts
index 1a0cb97..9e9939d 100644
--- a/src/compile.ts
+++ b/src/compile.ts
@@ -48,19 +48,10 @@ export function compileVueCode(code: string) {
       content = transferSetupPosition(content)
       content += '\nexport default _sfc_main\n'
 
-      if (scriptSetup.lang === 'ts' || scriptSetup.lang === 'tsx') {
-        ext = scriptSetup.lang
-      } else if (!scriptSetup.lang || scriptSetup.lang === 'js'|| scriptSetup.lang === 'jsx') {
-        ext = scriptSetup.lang || 'js'
-      }
+      ext = scriptSetup.lang || 'js'
     } else if (script && script.content) {
       content = script.content
-
-      if (script.lang === 'ts' || script.lang === 'tsx') {
-        ext = script.lang
-      } else if (!script.lang || script.lang === 'js' || script.lang === 'jsx') {
-        ext = script.lang || 'js'
-      }
+      ext = script.lang || 'js'
     }
   }
 
diff --git a/src/index.ts b/src/index.ts
index 2848f48..e381520 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -160,7 +160,7 @@ export default function dtsPlugin(options: PluginOptions = {}): Plugin {
         if (content) {
           if (ext === 'js' || ext === 'jsx') hasJsVue = true
 
-          project.createSourceFile(id + `.${ext || 'js'}`, content, { overwrite: true })
+          project.createSourceFile(`${id}.${ext || 'js'}`, content, { overwrite: true })
         }
       } else if (
         !id.includes('.vue?vue') &&