diff --git a/python/taichi/lang/__init__.py b/python/taichi/lang/__init__.py
index 2b82e5dbe98d9..124dd9fda6f75 100644
--- a/python/taichi/lang/__init__.py
+++ b/python/taichi/lang/__init__.py
@@ -28,11 +28,10 @@
 from taichi.lang.impl import (axes, deactivate_all_snodes, field, grouped,
                               ndarray, one, root, static, static_assert,
                               static_print, stop_grad, zero)
-from taichi.lang.kernel_arguments import SparseMatrixProxy
 from taichi.lang.kernel_impl import (KernelArgError, KernelDefError,
                                      data_oriented, func, kernel, pyfunc)
-from taichi.lang.matrix import Matrix, MatrixField, Vector
-from taichi.lang.mesh import Mesh, MeshElementFieldProxy, TetMesh, TriMesh
+from taichi.lang.matrix import *
+from taichi.lang.mesh import *
 from taichi.lang.ops import *  # pylint: disable=W0622
 from taichi.lang.quant_impl import quant
 from taichi.lang.runtime_ops import async_flush, sync
diff --git a/python/taichi/lang/matrix.py b/python/taichi/lang/matrix.py
index da697fc1d7cda..001c4e226312d 100644
--- a/python/taichi/lang/matrix.py
+++ b/python/taichi/lang/matrix.py
@@ -1429,3 +1429,6 @@ def fill_by_kernel(self, val):
 
     def __repr__(self):
         return f'<{self.n} {self.layout} ti.Vector.ndarray>'
+
+
+__all__ = ["Matrix", "Vector"]
diff --git a/python/taichi/lang/mesh.py b/python/taichi/lang/mesh.py
index e6790342b87c6..b999ead9429ae 100644
--- a/python/taichi/lang/mesh.py
+++ b/python/taichi/lang/mesh.py
@@ -528,3 +528,6 @@ def subscript(self, *indices):
         entry_expr.type_check()
         return MeshElementFieldProxy(self.mesh, self.to_element_type,
                                      entry_expr)
+
+
+__all__ = ["Mesh", "TetMesh", "TriMesh"]