Parallel imputation
small fix