diff --git a/ets2panda/linter/homecheck/src/checker/migration/CustomBuilderCheck.ts b/ets2panda/linter/homecheck/src/checker/migration/CustomBuilderCheck.ts
index b8dcb06db7d2c47226110a65dfc9298937fdc42c..8aa70c48075bda8fbe1c47aa070432687ee3721a 100644
--- a/ets2panda/linter/homecheck/src/checker/migration/CustomBuilderCheck.ts
+++ b/ets2panda/linter/homecheck/src/checker/migration/CustomBuilderCheck.ts
@@ -149,15 +149,18 @@ export class CustomBuilderCheck implements BaseChecker {
     }
 
     private isPassToCustomBuilder(stmt: Stmt, locals: Set<Local>): Local | undefined {
+        let res: Local | undefined = undefined;
         if (stmt instanceof ArkAssignStmt) {
-            if (!this.isCustomBuilderTy(stmt.getLeftOp().getType())) {
-                return undefined;
-            }
-            const rightOp = stmt.getRightOp();
-            if (rightOp instanceof Local && locals.has(rightOp)) {
-                return rightOp;
+            if (this.isCustomBuilderTy(stmt.getLeftOp().getType())) {
+                const rightOp = stmt.getRightOp();
+                if (rightOp instanceof Local && locals.has(rightOp)) {
+                    res = rightOp;
+                }
             }
         }
+        if (res !== undefined) {
+            return res;
+        }
         const invokeExpr = stmt.getInvokeExpr();
         if (invokeExpr) {
             const paramTys = invokeExpr.getMethodSignature().getMethodSubSignature().getParameterTypes();