diff --git a/python/pyspark/mllib/linalg.py b/python/pyspark/mllib/linalg.py
index cc9a4cf8ba1707edc59b6d0422d0d69882f6ffea..a57c0b3ae0d00dfea4c24fc1d06716c5c6103cbf 100644
--- a/python/pyspark/mllib/linalg.py
+++ b/python/pyspark/mllib/linalg.py
@@ -39,7 +39,8 @@ from pyspark.sql.types import UserDefinedType, StructField, StructType, ArrayTyp
     IntegerType, ByteType
 
 
-__all__ = ['Vector', 'DenseVector', 'SparseVector', 'Vectors', 'DenseMatrix', 'Matrices']
+__all__ = ['Vector', 'DenseVector', 'SparseVector', 'Vectors',
+           'Matrix', 'DenseMatrix', 'SparseMatrix', 'Matrices']
 
 
 if sys.version_info[:2] == (2, 7):