diff --git a/src/training/data.py b/src/training/data.py
index 12753d4..80d8226 100644
--- a/src/training/data.py
+++ b/src/training/data.py
@@ -444,7 +444,7 @@ def get_dataset_fn(data_path, dataset_type):
                 return get_wds_dataset
         else:
             raise ValueError(
-                f"Tried to figure out dataset type, but failed for extention {ext}.")
+                f"Tried to figure out dataset type, but failed for extension {ext}.")
     else:
         raise ValueError(f"Unsupported dataset type: {dataset_type}")