fix(babel): transform all arrow functions with block syntax

unjs · Mar 28, 2023 · ca052d6 · ca052d6
1 parent afa8dac
commit ca052d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/loader/babel.ts b/src/loader/babel.ts
@@ -30,9 +30,9 @@ const babelPluginUntyped: PluginItem = function (
           const newDeclaration = t.functionDeclaration(
             declaration.id,
             declaration.init.params,
-            t.isLiteral(declaration.init.body)
-              ? t.blockStatement([t.returnStatement(declaration.init.body)])
-              : (declaration.init.body as t.BlockStatement)
+            t.isBlockStatement(declaration.init.body)
+              ? (declaration.init.body as t.BlockStatement)
+              : t.blockStatement([t.returnStatement(declaration.init.body)])
           );
           newDeclaration.returnType = declaration.init.returnType;
           p.replaceWith(newDeclaration);