hi @khteh
the way you are predicting the death using the control estimator and treatment estimator is incorrect. you are just using the dataframe, where you are suppose to use the dataframe with the condition if death occurs in those conditions.
here is a thread i explained recently about the same which should help you understand your code issue
understanding condition column position
let me know if you still have issue.
regards
DP