diff --git a/evaluate_calibration.py b/evaluate_calibration.py
index 6422703..73eb210 100644
--- a/evaluate_calibration.py
+++ b/evaluate_calibration.py
@@ -1755,7 +1755,7 @@ def main() -> None:
     if landmark_ages.size == 0:
         raise ValueError("No landmark ages selected.")
 
-    death_token_ids = _get_death_token_ids(dataset, labels_meta)
+    death_token_ids = _get_death_token_ids(dataset)
     landmark_dataset = LandmarkDataset(
         dataset=dataset,
         subset_indices=subset_indices,