diff --git a/OBIA/OBIABase.py b/OBIA/OBIABase.py
index 11b81109711a7002d8517221edc1e8aeab677b9f..964b01c3fd601ba072fd16f3a01aabee76b77b98 100644
--- a/OBIA/OBIABase.py
+++ b/OBIA/OBIABase.py
@@ -319,6 +319,7 @@ class OBIABase:
         for tilenum in self.tiles.keys():
             tile_data = self.get_full_stats_on_tile(tilenum)
             L = tile_data.index.to_numpy(dtype=np.int32)
+            # Probably sub-optimal and error-prone, uses mean value from tile only 
             imp = SimpleImputer(missing_values=np.nan, strategy='mean')
             X = imp.fit_transform(tile_data[vars].to_numpy())
             if normalize is not None: