diff --git a/examples/tf/trpo_swimmer_ray_sampler.py b/examples/tf/trpo_swimmer_ray_sampler.py
index 146f2b8cff..4a1cb33b31 100755
--- a/examples/tf/trpo_swimmer_ray_sampler.py
+++ b/examples/tf/trpo_swimmer_ray_sampler.py
@@ -31,7 +31,7 @@ def trpo_swimmer_ray_sampler(ctxt=None, seed=1):
     """
     # Since this is an example, we are running ray in a reduced state.
     # One can comment this line out in order to run ray at full capacity
-    ray.init(memory=52428800,
+    ray.init(_memory=52428800,
              object_store_memory=78643200,
              ignore_reinit_error=True,
              log_to_driver=False,
diff --git a/examples/torch/trpo_pendulum_ray_sampler.py b/examples/torch/trpo_pendulum_ray_sampler.py
index cc7e9a5f96..3b8f4d142b 100755
--- a/examples/torch/trpo_pendulum_ray_sampler.py
+++ b/examples/torch/trpo_pendulum_ray_sampler.py
@@ -31,7 +31,7 @@ def trpo_pendulum_ray_sampler(ctxt=None, seed=1):
     """
     # Since this is an example, we are running ray in a reduced state.
     # One can comment this line out in order to run ray at full capacity
-    ray.init(memory=52428800,
+    ray.init(_memory=52428800,
              object_store_memory=78643200,
              ignore_reinit_error=True,
              log_to_driver=False,
diff --git a/tests/fixtures/sampler/ray_fixtures.py b/tests/fixtures/sampler/ray_fixtures.py
index 7a9f13a06c..affed63b21 100644
--- a/tests/fixtures/sampler/ray_fixtures.py
+++ b/tests/fixtures/sampler/ray_fixtures.py
@@ -34,7 +34,7 @@ def ray_session_fixture():
 
     """
     if not ray.is_initialized():
-        ray.init(memory=52428800,
+        ray.init(_memory=52428800,
                  object_store_memory=78643200,
                  ignore_reinit_error=True,
                  log_to_driver=False,