diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py
index 891ebba936bdf..be5827e079b5f 100644
--- a/python/pyspark/ml/regression.py
+++ b/python/pyspark/ml/regression.py
@@ -466,7 +466,7 @@ def _create_model(self, java_model):
 
 
 @inherit_doc
-class DecisionTreeModel(JavaModel, MLWritable, MLReadable):
+class DecisionTreeModel(JavaModel):
     """Abstraction for Decision Tree models.
 
     .. versionadded:: 1.5.0
@@ -506,7 +506,7 @@ def __repr__(self):
 
 
 @inherit_doc
-class DecisionTreeRegressionModel(DecisionTreeModel):
+class DecisionTreeRegressionModel(DecisionTreeModel, MLWritable, MLReadable):
     """
     Model fitted by DecisionTreeRegressor.