diff --git a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py
index 0432be490..2fde7c3d0 100644
--- a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py
+++ b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py
@@ -30,7 +30,7 @@ def load_hf_dataset(dataset_def: Dataset):
 
         dataset = hf_datasets.Dataset.from_pandas(df)
 
-    # drop rows not specified by schema
+    # drop columns not specified by schema
     if dataset_def.dataset_schema:
         dataset = dataset.select_columns(list(dataset_def.dataset_schema.keys()))