diff --git a/taichi/ir/frontend_ir.cpp b/taichi/ir/frontend_ir.cpp
index b29f0f44d6fdf..4fadbf195ea2e 100644
--- a/taichi/ir/frontend_ir.cpp
+++ b/taichi/ir/frontend_ir.cpp
@@ -38,7 +38,7 @@ FrontendSNodeOpStmt::FrontendSNodeOpStmt(ASTBuilder *builder,
 FrontendReturnStmt::FrontendReturnStmt(ASTBuilder *builder,
                                        const ExprGroup &group) {
   values = group;
-  auto expanded_exprs = builder->expand_expr(values.exprs);
+  auto expanded_exprs = builder->expand_exprs(values.exprs);
   values.exprs = std::move(expanded_exprs);
 }