diff --git a/tsdart/model.py b/tsdart/model.py
index 832389e..1e67ca7 100644
--- a/tsdart/model.py
+++ b/tsdart/model.py
@@ -370,7 +370,7 @@ def fit(self, data):
 
         Returns
         -------
-        TSDARTEstimator
+        self : TSDARTEstimator
         """
 
         states = self._model.transform(data, return_type='states')