diff --git a/python/taichi/lang/impl.py b/python/taichi/lang/impl.py
index 632cbc0e95d42..027696989dc9d 100644
--- a/python/taichi/lang/impl.py
+++ b/python/taichi/lang/impl.py
@@ -39,7 +39,7 @@ def expr_init_local_tensor(shape, element_type, elements):
 
 @taichi_scope
 def make_matrix_expr(shape, element_type, elements):
-    return get_runtime().prog.current_ast_builder().make_local_matrix(
+    return get_runtime().prog.current_ast_builder().make_matrix_expr(
         shape, element_type, elements)
 
 
diff --git a/taichi/python/export_lang.cpp b/taichi/python/export_lang.cpp
index 89ad1eaeb2a4f..8b44098656b3d 100644
--- a/taichi/python/export_lang.cpp
+++ b/taichi/python/export_lang.cpp
@@ -291,7 +291,7 @@ void export_lang(py::module &m) {
       .def("insert_deactivate", &ASTBuilder::insert_snode_deactivate)
       .def("insert_activate", &ASTBuilder::insert_snode_activate)
       .def("insert_external_func_call", &ASTBuilder::insert_external_func_call)
-      .def("make_local_matrix", &ASTBuilder::make_matrix_expr)
+      .def("make_matrix_expr", &ASTBuilder::make_matrix_expr)
       .def("expr_alloca", &ASTBuilder::expr_alloca)
       .def("expr_alloca_local_tensor", &ASTBuilder::expr_alloca_local_tensor)
       .def("expr_alloca_shared_array", &ASTBuilder::expr_alloca_shared_array)