diff --git a/src/cehrbert/data_generators/hf_data_generator/hf_dataset_mapping.py b/src/cehrbert/data_generators/hf_data_generator/hf_dataset_mapping.py
index 2714e73..8056572 100644
--- a/src/cehrbert/data_generators/hf_data_generator/hf_dataset_mapping.py
+++ b/src/cehrbert/data_generators/hf_data_generator/hf_dataset_mapping.py
@@ -30,7 +30,7 @@
     "Visit/61",
     "NUCC/315D00000X",
 ]
-ED_VISIT_TYPE_CODES = ["VISIT/ER"]
+ED_VISIT_TYPE_CODES = ["Visit/ER"]
 DISCHARGE_FACILITY_TYPES = [
     "8536",
     "8863",