diff --git a/docs/framework/operators/machine-learning/tree-regressor/README.md b/docs/framework/operators/machine-learning/tree-regressor/README.md
index 48b2dc1bc..e2e69b355 100644
--- a/docs/framework/operators/machine-learning/tree-regressor/README.md
+++ b/docs/framework/operators/machine-learning/tree-regressor/README.md
@@ -1,7 +1,6 @@
 # Tree Regressor
 
-`TreeRegressorTrait` provides a trait definition for decision tree regression.
-This trait offers functionalities to build a decision tree and predict target values based on input features.
+`TreeRegressorTrait` provides a trait definition for decision tree regression. This trait offers functionalities to build a decision tree and predict target values based on input features.
 
 ```rust
 use orion::operators::ml::TreeRegressorTrait;
@@ -21,4 +20,3 @@ Orion supports currently only fixed point data types for `TreeRegressorTrait`.
 | --- | --- |
 | [`tree.fit`](tree.fit.md) | Constructs a decision tree regressor based on the provided data and target values. |
 | [`tree.predict`](tree.predict.md) | Given a set of features, predicts the target value using the constructed decision tree. |
-
diff --git a/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md b/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md
index 28d4a027c..0d14f381a 100644
--- a/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md
+++ b/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md
@@ -1,6 +1,6 @@
-# TreeRegressorTrait::predict
+# tree.predict
 
-```rust 
+```rust
    fn predict(ref self: TreeNode<T>, features: Span<T>) -> T;
 ```