diff --git a/bluemath_tk/datamining/som.py b/bluemath_tk/datamining/som.py
index c356750..034cc07 100644
--- a/bluemath_tk/datamining/som.py
+++ b/bluemath_tk/datamining/som.py
@@ -370,7 +370,7 @@ def predict(self, data: pd.DataFrame) -> Tuple[np.ndarray, pd.DataFrame]:
             normalized_data=normalized_data.values
         )
 
-        return winner_neurons, self.centroids.iloc[winner_neurons]
+        return winner_neurons, self.centroids.loc[winner_neurons]
 
     def fit_predict(
         self,