diff --git a/utils/reloc_utils.py b/utils/reloc_utils.py
index 8cb965b..8e0658b 100644
--- a/utils/reloc_utils.py
+++ b/utils/reloc_utils.py
@@ -9,4 +9,5 @@ for n in range(N_max):
         binoms[n, k] = math.comb(n, k)
 
 def compute_relocation_cuda(opacity_old, scale_old, N):
+    N.clamp_(min=1, max=N_max-1)
     return compute_relocation(opacity_old, scale_old, N, binoms, N_max)
\ No newline at end of file