diff --git a/pyhealth/datasets/configs/mimic4_ehr.yaml b/pyhealth/datasets/configs/mimic4_ehr.yaml
index 0a5f8c7a1..3dfb0e5e5 100644
--- a/pyhealth/datasets/configs/mimic4_ehr.yaml
+++ b/pyhealth/datasets/configs/mimic4_ehr.yaml
@@ -32,6 +32,7 @@ tables:
     patient_id: "subject_id"
     timestamp: "intime"
     attributes:
+      - "hadm_id"
       - "stay_id"
       - "first_careunit"
       - "last_careunit"
@@ -96,6 +97,7 @@ tables:
           - "category"
     timestamp: "charttime"
     attributes:
+      - "hadm_id"
       - "itemid"
       - "label"
       - "fluid"