diff --git a/doc/source/ray-core/doc_code/pattern_async_actor.py b/doc/source/ray-core/doc_code/pattern_async_actor.py
index b45198c6049c..0f171f74df10 100644
--- a/doc/source/ray-core/doc_code/pattern_async_actor.py
+++ b/doc/source/ray-core/doc_code/pattern_async_actor.py
@@ -16,7 +16,7 @@ def __init__(self, task_store):
 
     def run(self):
         while True:
-            task = ray.get(task_store.get_next_task.remote())
+            task = ray.get(self.task_store.get_next_task.remote())
             self._execute_task(task)
 
     def _execute_task(self, task):
@@ -51,7 +51,7 @@ async def run(self):
             # Here we use await instead of ray.get() to
             # wait for the next task and it will yield
             # the control while waiting.
-            task = await task_store.get_next_task.remote()
+            task = await self.task_store.get_next_task.remote()
             self._execute_task(task)
 
     def _execute_task(self, task):