diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py
index 26ad85422..36f75d7b3 100644
--- a/llama_stack/apis/datasets/datasets.py
+++ b/llama_stack/apis/datasets/datasets.py
@@ -93,7 +93,7 @@ class CommonDatasetFields(BaseModel):
     Common fields for a dataset.
     """
     purpose: DatasetPurpose
-    data_source: DataSource
+    source: DataSource
     metadata: Dict[str, Any] = Field(
         default_factory=dict,
         description="Any additional metadata for this dataset",