diff --git a/docs/assets/report.png b/docs/assets/report.png
index 65e585b..6c86a95 100644
Binary files a/docs/assets/report.png and b/docs/assets/report.png differ
diff --git a/pyproject.toml b/pyproject.toml
index e793537..4993079 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "awblib"
-version = "0.0.2"
+version = "0.0.3"
 description = "A bunch of Automatic White-Balancing (AWB) Algorithm implementations"
 authors = ["Joey Teng <joey.teng.dev@gmail.com>"]
 license = "Apache-2.0"
diff --git a/src/awblib/chs.py b/src/awblib/chs.py
index 3045c26..c61204c 100644
--- a/src/awblib/chs.py
+++ b/src/awblib/chs.py
@@ -87,10 +87,15 @@ def original_chs(
     )
     assert isinstance(high, Integer[Array, "3"])
 
+    _max: Integer[Array, "3"] = RGB_range[:, 1].astype(int)
+    _min: Integer[Array, "3"] = RGB_range[:, 0].astype(int)
+    assert isinstance(_max, Integer[Array, "3"])
+    assert isinstance(_min, Integer[Array, "3"])
+
     # STAGE: Histogram Stretching
     result: RGBImage
-    result = ((image - low) * (RGB_range[:, 1]) / (high - low) +
-              RGB_range[:, 0]).astype(int)
+    result = ((image - low) * _max / (high - low) + _min).astype(int)
+    result = jnp.clip(result, a_min=0, a_max=max_value - 1).astype(int)
     assert isinstance(result, RGBImage)
 
     return result
@@ -156,14 +161,15 @@ def modified_chs(
     )
     assert isinstance(high, Integer[Array, "3"])
 
-    a_max: Integer[Array, ""] = high.max()  # A_max
-    a_min: Integer[Array, ""] = low.min()  # A_min
+    a_max: Integer[Array, ""] = high.max().astype(int)  # A_max
+    a_min: Integer[Array, ""] = low.min().astype(int)  # A_min
     assert isinstance(a_max, Integer[Array, ""])
     assert isinstance(a_min, Integer[Array, ""])
 
     # STAGE: Histogram Stretching
     result: RGBImage
-    result = ((image - low) * (a_max) / (high - low) + a_min).astype(int)
+    result = ((image - low) * a_max / (high - low) + a_min).astype(int)
+    result = jnp.clip(result, a_min=0, a_max=max_value - 1).astype(int)
     assert isinstance(result, RGBImage)
 
     return result
@@ -184,7 +190,7 @@ def balance(
 
     result: RGBImage
     if modified:
-        result = modified_chs(image)
+        result = modified_chs(image, bit_depth=bit_depth)
     else:
         result = original_chs(image, bit_depth=bit_depth, RGB_range=RGB_range)
     assert isinstance(result, RGBImage)