diff --git a/toolkit.py b/toolkit.py
index e25b458e90c62f4e278f31f7a3793565220487e3..52b911f2ad273c582002a4afcba1eb878aeb4d32 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -775,7 +775,7 @@ class ClassificationProject(object):
             class_weight = self.balanced_class_weight
         if self._w_train_tot is None:
             self._w_train_tot = self.w_train*np.array(class_weight)[self.y_train.astype(int)]
-            #self._w_train_tot /= np.mean(self._w_train_tot)
+            self._w_train_tot /= np.mean(self._w_train_tot)
         return self._w_train_tot