diff --git a/applications/rag/example_notebooks/rag-kaggle-ray-sql-interactive.ipynb b/applications/rag/example_notebooks/rag-kaggle-ray-sql-interactive.ipynb
index dac3b86fd..adb2fed26 100644
--- a/applications/rag/example_notebooks/rag-kaggle-ray-sql-interactive.ipynb
+++ b/applications/rag/example_notebooks/rag-kaggle-ray-sql-interactive.ipynb
@@ -229,7 +229,7 @@
     "# Distributed map batches to create chunks out of each row, and fetch the vector embeddings by running inference on the sentence transformer\n",
     "ds_embed = ds_batch.map_batches(\n",
     "    Embed,\n",
-    "    compute=ray.data.ActorPoolStrategy(size=ACTOR_POOL_SIZE),\n",
+    "    concurrency=ACTOR_POOL_SIZE,\n",
     "    batch_size=BATCH_SIZE,  # Large batch size to maximize GPU utilization.\n",
     "    num_gpus=1,  # 1 GPU for each actor.\n",
     "    # num_cpus=1,\n",