diff --git a/godot_rl/wrappers/ray_wrapper.py b/godot_rl/wrappers/ray_wrapper.py
index ab86cd1a..0562c57e 100644
--- a/godot_rl/wrappers/ray_wrapper.py
+++ b/godot_rl/wrappers/ray_wrapper.py
@@ -127,7 +127,7 @@ def rllib_training(args, extras):
             checkpoint_freq=checkpoint_freq,
             checkpoint_at_end=not args.eval,
             restore=args.restore,
-            local_dir=os.path.abspath(args.experiment_dir) or os.path.abspath("logs/rllib"),
+            storage_path=os.path.abspath(args.experiment_dir) or os.path.abspath("logs/rllib"),
             trial_name_creator=lambda trial: (
                 f"{args.experiment_name}" if args.experiment_name else f"{trial.trainable_name}_{trial.trial_id}"
             ),