diff --git a/bdpy/ml/learning.py b/bdpy/ml/learning.py
index d83a6ff7..f80dea9e 100644
--- a/bdpy/ml/learning.py
+++ b/bdpy/ml/learning.py
@@ -362,6 +362,7 @@ def run(self):
                     y_norm = self.Y_normalize['std']
                 Y = (Y - y_mean) / y_norm
                 Y[np.isinf(Y)] = 0
+                Y[np.isnan(Y)] = 0
 
             if not self.Y_sort is None:
                 print('Sorting Y')