diff --git a/src/RL/RL.jl b/src/RL/RL.jl
index 25ba548..d72b284 100644
--- a/src/RL/RL.jl
+++ b/src/RL/RL.jl
@@ -55,7 +55,7 @@ end
 function run_rl(;
     EnvType::Type{E},
     parent_dir_appendix::String,
-    elliptical_a_b_ratio::Float64,
+    elliptical_a_b_ratio::Float64=1.0,
     n_episodes::Int64=200,
     episode_duration::Float64=50.0,
     update_actions_at::Float64=0.1,
@@ -84,7 +84,7 @@ function run_rl(;
     )
     n_particles = sim_consts.n_particles # Not always equal to the input!
 
-    env_args = (skin_radius=sim_consts.skin_radius,)
+    env_args = (skin_radius=sim_consts.skin_radius, half_box_len=sim_consts.half_box_len)
 
     env = EnvType(; args=env_args)