diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py
index 8cd94442b..2ab958782 100644
--- a/llama_stack/apis/datasets/datasets.py
+++ b/llama_stack/apis/datasets/datasets.py
@@ -49,7 +49,7 @@ class Datasets(Protocol):
     async def register_dataset(
         self,
         dataset_id: str,
-        schema: Dict[str, ParamType],
+        dataset_schema: Dict[str, ParamType],
         url: URL,
         provider_dataset_id: Optional[str] = None,
         provider_id: Optional[str] = None,
diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py
index c039d3cb1..3345f4c26 100644
--- a/llama_stack/distribution/routers/routing_tables.py
+++ b/llama_stack/distribution/routers/routing_tables.py
@@ -310,7 +310,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
     async def register_dataset(
         self,
         dataset_id: str,
-        schema: Dict[str, ParamType],
+        dataset_schema: Dict[str, ParamType],
         url: URL,
         provider_dataset_id: Optional[str] = None,
         provider_id: Optional[str] = None,
@@ -332,7 +332,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
             identifier=dataset_id,
             provider_resource_id=provider_dataset_id,
             provider_id=provider_id,
-            dataset_schema=schema,
+            dataset_schema=dataset_schema,
             url=url,
             metadata=metadata,
         )