diff --git a/code/bolsonaro/models/omp_forest_classifier.py b/code/bolsonaro/models/omp_forest_classifier.py
index a86e53ba862a99379e92a9dc9ca6a688178164c5..31ca945b6b5aa4211dc924573a0751c90c4bae61 100644
--- a/code/bolsonaro/models/omp_forest_classifier.py
+++ b/code/bolsonaro/models/omp_forest_classifier.py
@@ -48,9 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest):
 
         result_omp = np.mean(omp_trees_predictions, axis=1)
 
-        result_omp = (result_omp - 0.5) * 2
-
-        print(result_omp)
+        result_omp = np.array([-1 if pred < 0.5 else 1 for pred in result_omp])
 
         return result_omp