diff --git a/src/common/lut.py b/src/common/lut.py
index 3c7a4b6..9791197 100644
--- a/src/common/lut.py
+++ b/src/common/lut.py
@@ -12,7 +12,7 @@ class Domain4DValues(Dataset):
     def __init__(self, quantization_interval=1):
         super(Domain4DValues, self).__init__()
         values1d = torch.arange(0, 256, quantization_interval, dtype=torch.uint8)
-        values1d = torch.cat([values1d, torch.tensor([255])])
+        values1d = torch.cat([values1d, torch.tensor([256])])
         self.quantization_interval = quantization_interval
         self.values = torch.cartesian_prod(*([values1d]*4)).view(-1, 1, 2, 2)