diff --git a/src/core/glint_removal/Hedley.py b/src/core/glint_removal/Hedley.py
index f2da2c3..edac624 100644
--- a/src/core/glint_removal/Hedley.py
+++ b/src/core/glint_removal/Hedley.py
@@ -23,7 +23,7 @@ class Hedley:
         """
         self.im_aligned = im_aligned
         self.bbox = self._read_shp_to_bbox(shp_path) if shp_path else None
-        self.NIR_band = NIR_band
+        self.NIR_band = int(float(NIR_band))
         self.n_bands = im_aligned.shape[-1]
         self.height = im_aligned.shape[0]
         self.width = im_aligned.shape[1]