diff --git a/ray-operator/test/support/create_detached_actor.py b/ray-operator/test/support/create_detached_actor.py
index 8fa7b7ba327..9a5ab968798 100644
--- a/ray-operator/test/support/create_detached_actor.py
+++ b/ray-operator/test/support/create_detached_actor.py
@@ -9,9 +9,7 @@
 parser.add_argument('--num-custom-resources', type=float, default=0)
 args = parser.parse_args()
 
-# set max_restarts=-1 as a workaround to restart unexpected death in tests.
-# TODO (rueian): Remove the max_restarts workaround when https://github.com/ray-project/ray/issues/40864 is fixed.
-@ray.remote(max_restarts=-1, num_cpus=args.num_cpus, num_gpus=args.num_gpus, resources={"CustomResource": args.num_custom_resources})
+@ray.remote(num_cpus=args.num_cpus, num_gpus=args.num_gpus, resources={"CustomResource": args.num_custom_resources})
 class Actor:
     pass