diff --git a/src/RL/Envs/LocalCOMWithAdditionalShapeRewardEnv.jl b/src/RL/Envs/LocalCOMWithAdditionalShapeRewardEnv.jl
index 3181228..539e5a0 100644
--- a/src/RL/Envs/LocalCOMWithAdditionalShapeRewardEnv.jl
+++ b/src/RL/Envs/LocalCOMWithAdditionalShapeRewardEnv.jl
@@ -59,7 +59,9 @@ mutable struct LocalCOMWithAdditionalShapeRewardEnvHelper <: EnvHelper
     function LocalCOMWithAdditionalShapeRewardEnvHelper(
         shared::EnvHelperSharedProps, half_box_len::Float64, skin_radius::Float64
     )
-        max_elliptical_distance = sqrt(2) * half_box_len / shared.elliptical_a_b_ratio
+        max_elliptical_distance = sqrt(
+            half_box_len^2 + (half_box_len / shared.elliptical_a_b_ratio)^2
+        )
 
         max_distance_to_local_center_of_mass = skin_radius