fix imputation tool dependency