diff --git a/packages/vue-code-gen/src/generators/script.ts b/packages/vue-code-gen/src/generators/script.ts
index 11bdc1e7e..e820b705c 100644
--- a/packages/vue-code-gen/src/generators/script.ts
+++ b/packages/vue-code-gen/src/generators/script.ts
@@ -326,7 +326,7 @@ export function generate(
 
 		if (scriptRanges?.exportDefault && scriptRanges.exportDefault.expression.start !== scriptRanges.exportDefault.args.start) {
 			// use defineComponent() from user space code if it exist
-			codeGen.addText(`return `);
+			codeGen.addText(`const __VLS_Component = `);
 			addVirtualCode('script', scriptRanges.exportDefault.expression.start, scriptRanges.exportDefault.args.start);
 			codeGen.addText(`{\n`);
 		}