diff --git a/src/rl4caribou/envs/caribou.py b/src/rl4caribou/envs/caribou.py
index 92385ec..e4068ed 100644
--- a/src/rl4caribou/envs/caribou.py
+++ b/src/rl4caribou/envs/caribou.py
@@ -191,7 +191,7 @@ def state_units(self, pop):
     def population_units(self):
         pop = (self.state + 1) * self.bound / 2
         return np.clip(
-            pop, np.repeat(0, pop.__len__()), np.repeat(np.Inf, pop.__len__())
+            pop, np.repeat(0, len(pop)), np.repeat(None, len(pop))
         )
 
 
diff --git a/src/rl4caribou/envs/caribou_ode.py b/src/rl4caribou/envs/caribou_ode.py
index 6f2576a..b647481 100644
--- a/src/rl4caribou/envs/caribou_ode.py
+++ b/src/rl4caribou/envs/caribou_ode.py
@@ -274,7 +274,7 @@ def state_units(self, pop):
     def population_units(self):
         pop = (self.state + 1) * self.bound / 2
         return np.clip(
-            pop, np.repeat(0, pop.__len__()), np.repeat(np.Inf, pop.__len__())
+            pop, np.repeat(0, len(pop)), np.repeat(None, len(pop))
         )