diff --git a/src/cehrbert/spark_apps/decorators/patient_event_decorator.py b/src/cehrbert/spark_apps/decorators/patient_event_decorator.py
index 7eaf2409..de49ddd6 100644
--- a/src/cehrbert/spark_apps/decorators/patient_event_decorator.py
+++ b/src/cehrbert/spark_apps/decorators/patient_event_decorator.py
@@ -16,7 +16,7 @@ class AttType(Enum):
     DAY = "day"
     WEEK = "week"
     MONTH = "month"
-    CEHR_BERT = "cehrbert"
+    CEHR_BERT = "cehr_bert"
     MIX = "mix"
     NONE = "none"