zod-v3-to-v4 1.6.0 → 1.6.1

package/dist/convert-name-to-top-level-api.js

@@ -267,20 +267,17 @@ function convertNameToTopLevelApiAndWrapInUnion(node, options) {
     const { zodName, oldName, nameToWrap, renames } = options;
     const names = [nameToWrap];
     getCallExpressionsToConvert(node, { oldName: nameToWrap, names }).forEach((callExpression) => {
-        // Remove deprecated names from the chain
         callExpression
             .getDescendantsOfKind(SyntaxKind.PropertyAccessExpression)
             .filter((expression) => names.includes(expression.getName()))
             .forEach((expression) => {
             const parent = expression.getFirstAncestorByKind(SyntaxKind.CallExpression);
-            parent?.replaceWithText(expression.getExpression().getText());
+            const text = expression.getExpression().getText();
+            const unionText = renames
+                .map(({ name }) => `${text}.${name}()`)
+                .join(", ");
+            parent?.replaceWithText(`${zodName}.union([${unionText}])`);
         });
-        // Nest the whole expression inside a union for ipv4() & ipv6()
-        const text = callExpression.getText();
-        const unionText = renames
-            .map(({ name }) => `${text}.${name}()`)
-            .join(", ");
-        callExpression?.replaceWithText(`${zodName}.union([${unionText}])`);
         convertNameToTopLevelApi(callExpression, {
             zodName,
             oldName,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zod-v3-to-v4",
-  "version": "1.6.0",
+  "version": "1.6.1",
   "description": "Migrate Zod from v3 to v4",
   "keywords": [
     "zod",