diff --git a/fastvideo/dataset/dataloader/record_schema.py b/fastvideo/dataset/dataloader/record_schema.py
index ebb8cb5ea..d642d8ef9 100644
--- a/fastvideo/dataset/dataloader/record_schema.py
+++ b/fastvideo/dataset/dataloader/record_schema.py
@@ -72,7 +72,7 @@ def i2v_record_creator(batch: PreprocessBatch) -> list[dict[str, Any]]:
     """Create a record for the Parquet dataset with CLIP features."""
     records = basic_t2v_record_creator(batch)
 
-    # Adapt for model doesn't have image decoder, e.g., Hunyuan
+    # Adapt for model doesn't have image encoder, e.g., Hunyuan
     if len(batch.image_embeds) == 0:
         image_embeds = None
     elif len(batch.image_embeds) == 1: