diff --git a/r-analysis/EffectsOfEnrollmentDelay.qmd b/r-analysis/EffectsOfEnrollmentDelay.qmd
index 10d021d..87df034 100644
--- a/r-analysis/EffectsOfEnrollmentDelay.qmd
+++ b/r-analysis/EffectsOfEnrollmentDelay.qmd
@@ -803,7 +803,7 @@ pddf_ib <- data.frame(extract(generated_ib, pars="predicted_difference")$predict
     pivot_longer(X1:X168) #CHANGE_NOTE: moved from X169 to X168
 
 pddf_ib["entry_idx"] <- as.numeric(gsub("\\D","",pddf_ib$name))
-pddf_ib["category"] <-  sapply(pddf_ib$entry_idx, function(i) df$category_id[i])
+pddf_ib["category"] <-  sapply(pddf_ib$entry_idx, function(i) counterfact_delay$llx[i])
 pddf_ib["category_name"] <- sapply(
     pddf_ib$category, 
     function(i) category_names[i]