diff --git a/code/bolsonaro/models/ensemble_selection_forest_regressor.py b/code/bolsonaro/models/ensemble_selection_forest_regressor.py
index a20d8ec8bee5a124a10bf708ae176cf48796458e..50c1dc8570d04dcb4ee4825e67298f9ea79b861f 100644
--- a/code/bolsonaro/models/ensemble_selection_forest_regressor.py
+++ b/code/bolsonaro/models/ensemble_selection_forest_regressor.py
@@ -31,7 +31,7 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta):
             val_score = self._score_metric(estimator.predict(X_val), y_val)
             scores_list.append(val_score)
 
-        class_list = list(library)
+        class_list = list(self._library)
         m = np.argmax(np.asarray(scores_list))
         self._ensemble_selected = [class_list[m]]
         temp_pred = class_list[m].predict(X_val)