diff --git a/code/bolsonaro/models/kmeans_forest_regressor.py b/code/bolsonaro/models/kmeans_forest_regressor.py
index 591b653e16503a65fa6baf1ab83f2b014eab7093..551c6d878b8ff8445c3385493195a450a346c899 100644
--- a/code/bolsonaro/models/kmeans_forest_regressor.py
+++ b/code/bolsonaro/models/kmeans_forest_regressor.py
@@ -121,15 +121,4 @@ class KMeansForestClassifier(KmeansForest, metaclass=ABCMeta):
     def _get_best_tree_index(self, y_preds, y_true):
         score = self._score_metric(y_preds, y_true)
         best_tree_index = np.argmax(score)  # get best scoring tree (the one with lowest mse)
-        return best_tree_index
-
-if __name__ == "__main__":
-    from sklearn import datasets
-    from bolsonaro.models.model_parameters import ModelParameters
-
-    X, y = datasets.fetch_california_housing(return_X_y=True)
-    ModelParameters(extracted_forest_size=100,
-                    normalize_D=True,
-                    )
-
-    k_reg = KMeansForestRegressor()
\ No newline at end of file
+        return best_tree_index
\ No newline at end of file