diff --git a/code/bolsonaro/models/model_raw_results.py b/code/bolsonaro/models/model_raw_results.py
index fcb4220896e89e3a000f1058e34316dd9073a883..fc22406bcd6fc1d44023be79783961a32eccb172 100644
--- a/code/bolsonaro/models/model_raw_results.py
+++ b/code/bolsonaro/models/model_raw_results.py
@@ -22,7 +22,7 @@ class ModelRawResults(object):
         self._test_score_base = test_score_base
         self._score_metric = score_metric
         self._base_score_metric = base_score_metric
-    
+
     @property
     def model_weights(self):
         return self.model_weights
diff --git a/code/train.py b/code/train.py
index e51514cc254ee564993243a676b05d07e3aa7597..8a7ccebed390e318ad74019aed2e14d704fe67dd 100644
--- a/code/train.py
+++ b/code/train.py
@@ -208,7 +208,7 @@ if __name__ == "__main__":
     parameters['extracted_forest_size'] = np.unique(np.around(hyperparameters['n_estimators'] *
         np.linspace(0, args.extracted_forest_size_stop,
         parameters['extracted_forest_size_samples'] + 1,
-        endpoint=False)[1:]).astype(np.int)).tolist()
+        endpoint=True)[1:]).astype(np.int)).tolist()
 
     if parameters['seeds'] != None and parameters['random_seed_number'] > 1:
         logger.warning('seeds and random_seed_number parameters are both specified. Seeds will be used.')