diff --git a/pyproject.toml b/pyproject.toml
index 62859ee..2922258 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,7 +58,7 @@ dependencies = [
     "Werkzeug==3.0.1",
     "wandb==0.17.8",
     "xgboost==2.0.3",
-    "cehrbert_data==0.0.3"
+    "cehrbert_data==0.0.4"
 ]
 
 [tool.setuptools_scm]