diff --git a/code/bolsonaro/models/omp_forest_classifier.py b/code/bolsonaro/models/omp_forest_classifier.py
index 31ca945b6b5aa4211dc924573a0751c90c4bae61..a51405a6a3278bb86dd52d011b599175bbfc7482 100644
--- a/code/bolsonaro/models/omp_forest_classifier.py
+++ b/code/bolsonaro/models/omp_forest_classifier.py
@@ -48,7 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest):
 
         result_omp = np.mean(omp_trees_predictions, axis=1)
 
-        result_omp = np.array([-1 if pred < 0.5 else 1 for pred in result_omp])
+        result_omp = (result_omp - 0.5) * 2
 
         return result_omp
 
diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py
index 0e437d223ddabdac1e9a020a05e9a3691b6e3d0e..cb1702d0b7e49381fc4c881476a3d5797e1d31c4 100644
--- a/code/bolsonaro/trainer.py
+++ b/code/bolsonaro/trainer.py
@@ -108,7 +108,7 @@ class Trainer(object):
             else:
                 y_pred = model.predict_no_weights(X)
             if type(model) is OmpForestBinaryClassifier:
-                y_pred = y_pred.round()
+                y_pred = np.sign(y_pred)
             result = self._classification_score_metric(y_true, y_pred)
         return result