diff --git a/examples/rag_bm.py b/examples/rag_bm.py
index a67eddc7c..bffe1e49b 100644
--- a/examples/rag_bm.py
+++ b/examples/rag_bm.py
@@ -45,7 +45,7 @@ class RAGExample:
         self.llm = get_rag_llm()
 
     async def rag_evaluate_pipeline(self, dataset_name: list[str] = ["CRUD"]):
-        dataset_config = self.benchmark.load_dataset()
+        dataset_config = self.benchmark.load_dataset(dataset_name)
 
         for dataset in dataset_config.datasets:
             if "all" in dataset_name or dataset.name in dataset_name: