fix imputeMissing for multiple group models