diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index 784e19467437835593018c2e74237a994cff43be..fc5a4d966b79c202ae1964ce8a13033e665b4732 100644
--- a/mlair/data_handler/default_data_handler.py
+++ b/mlair/data_handler/default_data_handler.py
@@ -177,7 +177,7 @@ class DefaultDataHandler(AbstractDataHandler):
         X = self._X
         for i_bin in range(len(bin_edges)-1):
             bin_start = bin_edges[i_bin]
-            if i_bin == len(bin_edges) - 1:
+            if i_bin == len(bin_edges) - 2:
                 bin_end = bin_edges[i_bin+1]+1
             else:
                 bin_end = bin_edges[i_bin + 1]